tests (2)

INGEOTEC · Jan 30, 2024 · 8b61085 · 8b61085
1 parent 7e14fe8
commit 8b61085
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/IngeoML/tests/test_optimizer.py b/IngeoML/tests/test_optimizer.py
@@ -109,6 +109,7 @@ def modelo(params, X):
                       W0=jnp.array(m.intercept_))
     p, evol = classifier(parameters, modelo, X, y,
                          # learning_rate=1e-3,
+                         n_iter_no_change=10,
                          return_evolution=True)
     evol = np.array([x[1] for x in evol])
     assert np.any(np.diff(evol) != 0)
@@ -119,6 +120,7 @@ def modelo(params, X):
     parameters = dict(W=jnp.array(m.coef_.T),
                       W0=jnp.array(m.intercept_))
     p2, evol = classifier(parameters, modelo, X, y,
+                          n_iter_no_change=10,
                           return_evolution=True)
     evol = np.array([x[1] for x in evol])
     assert np.any(np.diff(evol) != 0)
@@ -143,6 +145,7 @@ def modelo(params, X, X2):
                       W0=jnp.array(m.intercept_))
     p, evol = classifier(parameters, modelo, X, y,
                          # learning_rate=1e-3,
+                         n_iter_no_change=10,
                          return_evolution=True,
                          model_args=(X,))
     evol = np.array([x[1] for x in evol])
@@ -172,6 +175,7 @@ def initial_parameters(X, y, X2):
     X, y = load_wine(return_X_y=True)
     p, evol = classifier(initial_parameters, modelo, X, y,
                          return_evolution=True,
+                         n_iter_no_change=10,
                          model_args=(X,))
     evol = np.array([x[1] for x in evol])
     assert np.any(np.diff(evol) != 0)