summaryrefslogtreecommitdiff
path: root/examples/plot_OT_2D_samples.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2017-06-13 15:50:11 +0200
committerRémi Flamary <remi.flamary@gmail.com>2017-06-13 15:50:11 +0200
commit38e96f88eb520b9fa8333686565b082d2921e131 (patch)
treea0644f3fe062e9f6b8ba4c279934b0311508d78c /examples/plot_OT_2D_samples.py
parenta7bed093f91922e18fa5902c4d1d63b9712d5794 (diff)
implement paralell sinkhorn stabilized
Diffstat (limited to 'examples/plot_OT_2D_samples.py')
-rw-r--r--examples/plot_OT_2D_samples.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/plot_OT_2D_samples.py b/examples/plot_OT_2D_samples.py
index 3b95083..edfb781 100644
--- a/examples/plot_OT_2D_samples.py
+++ b/examples/plot_OT_2D_samples.py
@@ -13,7 +13,7 @@ import ot
#%% parameters and data generation
-n=2 # nb samples
+n=50 # nb samples
mu_s=np.array([0,0])
cov_s=np.array([[1,0],[0,1]])
@@ -62,7 +62,7 @@ pl.title('OT matrix with samples')
#%% sinkhorn
# reg term
-lambd=5e-3
+lambd=5e-4
Gs=ot.sinkhorn(a,b,M,lambd)