|
|
| __init__ (self, np.ndarray x, float step, Callable|None grad=None, Callable|None prox=None, float t0=1, np.ndarray|None z0=None) |
| |
| | update (self, int it, np.ndarray input_grad, *args) |
| |
| FistaParameter | __deepcopy__ (self, dict[int, Any]|None memo=None) |
| |
| FistaParameter | __copy__ (self) |
| |
|
| __init__ (self, np.ndarray x, dict[str, np.ndarray] helpers, Callable|float step, Callable|None grad=None, Callable|None prox=None) |
| |
| float | step (self) |
| |
| tuple[int,...] | shape (self) |
| |
| npt.DTypeLike | dtype (self) |
| |
| Parameter | __copy__ (self) |
| |
| Parameter | __deepcopy__ (self, dict[int, Any]|None memo=None) |
| |
| Parameter | copy (self, bool deep=False) |
| |
| | resize (self, Box old_box, Box new_box) |
| |
A `Parameter` that updates itself using the Beck-Teboulle 2009
FISTA proximal gradient method.
See https://www.ceremade.dauphine.fr/~carlier/FISTA
Parameters
----------
x:
The array of values that is being fit.
step:
A numerical step value or function to calculate the step for a
given `x`.
grad:
A function to calculate the gradient of `x`.
prox:
A function to take the proximal operator of `x`.
t0:
The initial value of the FISTA momentum term.
z0:
The initial value of the extrapolation array.
If `None` then a copy of `x` is used.