Skip to content

Commit c82e32e

Browse files
committed
changed obj_freq from 100 to 10
1 parent 8584299 commit c82e32e

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

skglm/solvers/fista.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55

66
@njit
77
def _prox_vec(w, z, penalty, lipschitz):
8-
# XXX: TO DISCUSS: should add a vectorized prox update
98
n_features = w.shape[0]
109
for j in range(n_features):
1110
w[j] = penalty.prox_1d(z[j], 1 / lipschitz, j)
@@ -16,7 +15,7 @@ class FISTA(BaseSolver):
1615
r"""ISTA solver with Nesterov acceleration (FISTA)."""
1716

1817
def __init__(self, max_iter=100, tol=1e-4, fit_intercept=False, warm_start=False,
19-
opt_freq=100, verbose=0):
18+
opt_freq=10, verbose=0):
2019
self.max_iter = max_iter
2120
self.tol = tol
2221
self.fit_intercept = fit_intercept

0 commit comments

Comments
 (0)