diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2018-09-24 10:29:37 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-09-24 10:29:37 +0200 |
commit | c9b99df8fffec1dcc6802ef43b6192774817c5fb (patch) | |
tree | 22939513930c1dd3c28fe93d90f2a7a284a0f82f /test/test_stochastic.py | |
parent | 4367a343aeb0ceccbb99acc0f92797af020bb537 (diff) | |
parent | ccbe274fd9554492bb88ddaf530c2800a8dc3418 (diff) |
Merge pull request #64 from rflamary/convolution
[MRG] Wasserstein convolutional barycenter
This PR closes Issue #51
Diffstat (limited to 'test/test_stochastic.py')
-rw-r--r-- | test/test_stochastic.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_stochastic.py b/test/test_stochastic.py index 0128317..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 = 300000 + nb_iter = 100000 rng = np.random.RandomState(0) x = rng.randn(n, 2) @@ -167,7 +167,7 @@ def test_dual_sgd_sinkhorn(): # test all dual algorithms n = 10 reg = 1 - nb_iter = 150000 + nb_iter = 15000 batch_size = 10 rng = np.random.RandomState(0) |