summaryrefslogtreecommitdiff
path: root/examples/plot_gromov_barycenter.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2017-09-15 12:41:02 +0200
committerRémi Flamary <remi.flamary@gmail.com>2017-09-15 12:41:02 +0200
commitbad3d95523d005a4fbf64dd009c716b9dd560fe3 (patch)
tree250d88fc139b0fce2f05892a0810b8b1917805b8 /examples/plot_gromov_barycenter.py
parentd27e14cb4a48e76ab132c53aca56155e9be50bda (diff)
clean nico commit mess with its emal
Diffstat (limited to 'examples/plot_gromov_barycenter.py')
-rwxr-xr-xexamples/plot_gromov_barycenter.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/examples/plot_gromov_barycenter.py b/examples/plot_gromov_barycenter.py
index 52f4966..93533c0 100755
--- a/examples/plot_gromov_barycenter.py
+++ b/examples/plot_gromov_barycenter.py
@@ -3,6 +3,7 @@
=====================================
Gromov-Wasserstein Barycenter example
=====================================
+
This example is designed to show how to use the Gromov-Wasserstein distance
computation in POT.
"""
@@ -24,7 +25,6 @@ from sklearn.decomposition import PCA
import ot
"""
-
Smacof MDS
==========
This function allows to find an embedding of points given a dissimilarity matrix
@@ -134,28 +134,28 @@ for i in range(2):
Ct01[i] = ot.gromov.gromov_barycenters(n_samples, [Cs[0], Cs[1]],
[ps[0], ps[1]
], p, lambdast[i], 'square_loss', 5e-4,
- max_iter=100, stopThr=1e-3)
+ max_iter=100, tol=1e-3)
Ct02 = [0 for i in range(2)]
for i in range(2):
Ct02[i] = ot.gromov.gromov_barycenters(n_samples, [Cs[0], Cs[2]],
[ps[0], ps[2]
], p, lambdast[i], 'square_loss', 5e-4,
- max_iter=100, stopThr=1e-3)
+ max_iter=100, tol=1e-3)
Ct13 = [0 for i in range(2)]
for i in range(2):
Ct13[i] = ot.gromov.gromov_barycenters(n_samples, [Cs[1], Cs[3]],
[ps[1], ps[3]
], p, lambdast[i], 'square_loss', 5e-4,
- max_iter=100, stopThr=1e-3)
+ max_iter=100, tol=1e-3)
Ct23 = [0 for i in range(2)]
for i in range(2):
Ct23[i] = ot.gromov.gromov_barycenters(n_samples, [Cs[2], Cs[3]],
[ps[2], ps[3]
], p, lambdast[i], 'square_loss', 5e-4,
- max_iter=100, stopThr=1e-3)
+ max_iter=100, tol=1e-3)
"""
Visualization