diff options
author | Kilian Fatras <kilianfatras@dhcp-206-12-53-92.eduroam.wireless.ubc.ca> | 2018-06-21 17:18:26 -0700 |
---|---|---|
committer | Kilian Fatras <kilianfatras@dhcp-206-12-53-92.eduroam.wireless.ubc.ca> | 2018-06-21 17:18:26 -0700 |
commit | 6777ffd5c8457faac4467e58ba9edbcf2f86961b (patch) | |
tree | 286c1fdb3902031d47cc5e1bdefab97ab9c9da22 /test | |
parent | 7073e417bed151976c62fe20d1ba69abd30e7758 (diff) |
gave better step size ASGD & SAG
Diffstat (limited to 'test')
-rw-r--r-- | test/test_stochastic.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/test/test_stochastic.py b/test/test_stochastic.py index 3cb51ff..f315c88 100644 --- a/test/test_stochastic.py +++ b/test/test_stochastic.py @@ -63,7 +63,6 @@ def test_stochastic_asgd(): n = 15 reg = 1 numItermax = 300000 - lr = 1 rng = np.random.RandomState(0) x = rng.randn(n, 2) @@ -72,8 +71,7 @@ def test_stochastic_asgd(): M = ot.dist(x, x) G = ot.stochastic.solve_semi_dual_entropic(u, u, M, reg, "asgd", - numItermax=numItermax, - lr=lr) + numItermax=numItermax) # check constratints np.testing.assert_allclose( @@ -95,7 +93,6 @@ def test_sag_asgd_sinkhorn(): n = 15 reg = 1 nb_iter = 300000 - lr = 1 rng = np.random.RandomState(0) x = rng.randn(n, 2) @@ -104,7 +101,7 @@ def test_sag_asgd_sinkhorn(): M = ot.dist(x, x) G_asgd = ot.stochastic.solve_semi_dual_entropic(u, u, M, reg, "asgd", - numItermax=nb_iter, lr=lr) + numItermax=nb_iter) G_sag = ot.stochastic.solve_semi_dual_entropic(u, u, M, reg, "sag", numItermax=nb_iter) G_sinkhorn = ot.sinkhorn(u, u, M, reg) |