diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-14 09:03:07 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-14 09:03:07 +0200 |
| commit | ccbe274fd9554492bb88ddaf530c2800a8dc3418 (patch) | |
| tree | 4270e86e0c05ca6f0f4c2fe1ecf32828d6cee9b6 /test | |
| parent | 7786e8502ea010a52918861014ab0e7a934a7d45 (diff) | |
speedup test stochastic
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_stochastic.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_stochastic.py b/test/test_stochastic.py index 62cf82a..f0f3fc8 100644 --- a/test/test_stochastic.py +++ b/test/test_stochastic.py @@ -32,7 +32,7 @@ def test_stochastic_sag(): # test sag n = 15 reg = 1 - numItermax = 300000 + numItermax = 30000 rng = np.random.RandomState(0) x = rng.randn(n, 2) @@ -62,7 +62,7 @@ def test_stochastic_asgd(): # test asgd n = 15 reg = 1 - numItermax = 300000 + numItermax = 100000 rng = np.random.RandomState(0) x = rng.randn(n, 2) @@ -92,7 +92,7 @@ def test_sag_asgd_sinkhorn(): # test all algorithms n = 15 reg = 1 - nb_iter = 30000 + nb_iter = 100000 rng = np.random.RandomState(0) x = rng.randn(n, 2) |