diff options
| author | tvayer <titouan.vayer@gmail.com> | 2019-05-29 14:24:05 +0200 |
|---|---|---|
| committer | tvayer <titouan.vayer@gmail.com> | 2019-05-29 14:24:05 +0200 |
| commit | 63bbeb34e48f02c97a762dab5232158d90a5cffc (patch) | |
| tree | 853026b5854b6e4b01fdf750db139985b3dd596f /test/test_gromov.py | |
| parent | f70aabfcc11f92181e0dc987b341bad8ec030d75 (diff) | |
| parent | f66ab58c7c895011fd37bafd3e848828399c56c4 (diff) | |
Merge remote-tracking branch 'rflamary/master'
merge pot
Diffstat (limited to 'test/test_gromov.py')
| -rw-r--r-- | test/test_gromov.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/test/test_gromov.py b/test/test_gromov.py index 07cd874..43b63e1 100644 --- a/test/test_gromov.py +++ b/test/test_gromov.py @@ -28,7 +28,7 @@ def test_gromov(): C1 /= C1.max()
C2 /= C2.max()
- G = ot.gromov.gromov_wasserstein(C1, C2, p, q, 'square_loss')
+ G = ot.gromov.gromov_wasserstein(C1, C2, p, q, 'square_loss', verbose=True)
# check constratints
np.testing.assert_allclose(
@@ -69,7 +69,7 @@ def test_entropic_gromov(): C2 /= C2.max()
G = ot.gromov.entropic_gromov_wasserstein(
- C1, C2, p, q, 'square_loss', epsilon=5e-4)
+ C1, C2, p, q, 'square_loss', epsilon=5e-4, verbose=True)
# check constratints
np.testing.assert_allclose(
@@ -107,7 +107,8 @@ def test_gromov_barycenter(): [ot.unif(ns), ot.unif(nt)
], ot.unif(n_samples), [.5, .5],
'square_loss', # 5e-4,
- max_iter=100, tol=1e-3)
+ max_iter=100, tol=1e-3,
+ verbose=True)
np.testing.assert_allclose(Cb.shape, (n_samples, n_samples))
Cb2 = ot.gromov.gromov_barycenters(n_samples, [C1, C2],
@@ -134,7 +135,8 @@ def test_gromov_entropic_barycenter(): [ot.unif(ns), ot.unif(nt)
], ot.unif(n_samples), [.5, .5],
'square_loss', 2e-3,
- max_iter=100, tol=1e-3)
+ max_iter=100, tol=1e-3,
+ verbose=True)
np.testing.assert_allclose(Cb.shape, (n_samples, n_samples))
Cb2 = ot.gromov.entropic_gromov_barycenters(n_samples, [C1, C2],
|