diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2016-10-28 09:14:19 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2016-10-28 09:14:19 +0200 |
| commit | 0e8363ca13034537ca7bc64acd982f39b7a42123 (patch) | |
| tree | 18adc06132943c2c9d0ba731a33a9b7174624b2f /examples | |
| parent | cf2d92e151f816e6ddcfc4b64cbda1f8f7bde9df (diff) | |
doc sinkhorn
Diffstat (limited to 'examples')
| -rw-r--r-- | examples/demo_OT_1D.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/examples/demo_OT_1D.py b/examples/demo_OT_1D.py index fc2cbcf..6eaa2ff 100644 --- a/examples/demo_OT_1D.py +++ b/examples/demo_OT_1D.py @@ -8,7 +8,7 @@ Demo for 1D optimal transport import numpy as np import matplotlib.pylab as pl import ot - +from ot.datasets import get_1D_gauss as gauss #%% parameters @@ -19,8 +19,8 @@ n=100 # nb bins x=np.arange(n,dtype=np.float64) # Gaussian distributions -a=ot.datasets.get_1D_gauss(n,m=20,s=20) # m= mean, s= std -b=ot.datasets.get_1D_gauss(n,m=60,s=60) +a=gauss(n,m=20,s=20) # m= mean, s= std +b=gauss(n,m=60,s=60) # loss matrix M=ot.dist(x.reshape((n,1)),x.reshape((n,1))) |