From 8b61085f96377e8ac8b36ab4b623d7c6d7f446a9 Mon Sep 17 00:00:00 2001 From: Mario Graff Date: Tue, 30 Jan 2024 12:22:54 -0600 Subject: [PATCH] tests (2) --- IngeoML/tests/test_optimizer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/IngeoML/tests/test_optimizer.py b/IngeoML/tests/test_optimizer.py index 40f0a05..2ede108 100644 --- 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)