pysindy.utils.get_prox

pysindy.utils.get_prox(regularization: str) → Callable[[ndarray[tuple[Any, ...], dtype[float64]], float | ndarray[tuple[Any, ...], dtype[float64]]], ndarray[tuple[Any, ...], dtype[float64]]]

Args:

regularization: ‘l0’ | ‘weighted_l0’ | ‘l1’ | ‘weighted_l1’ | ‘l2’ | ‘weighted_l2’

Returns:

proximal_operator: (x: np.array, reg_weight: float | np.array) -> np.array

A function that takes an input array x and a regularization weight, which can be either a scalar or array of the same shape, and returns an array of the same shape