diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-06-11 12:01:27 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-06-11 12:01:27 +0200 |
commit | ef17fcd2d5e85b986ff21d8039483bdaf03e37da (patch) | |
tree | 25cf5915a26273b3e0726494ed4a1405ce1e2786 /docs/source/auto_examples/plot_OT_2D_samples.py | |
parent | 8046b8c424d7b80f520e212e2bf8de41cb624aab (diff) | |
parent | 530dc93a60e9b81fb8d1b44680deea77dacf660b (diff) |
Merge branch 'master' into smooth_ot
Diffstat (limited to 'docs/source/auto_examples/plot_OT_2D_samples.py')
-rw-r--r-- | docs/source/auto_examples/plot_OT_2D_samples.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/docs/source/auto_examples/plot_OT_2D_samples.py b/docs/source/auto_examples/plot_OT_2D_samples.py index 9818ec5..bb952a0 100644 --- a/docs/source/auto_examples/plot_OT_2D_samples.py +++ b/docs/source/auto_examples/plot_OT_2D_samples.py @@ -16,6 +16,7 @@ sum of diracs. The OT matrix is plotted with the samples. import numpy as np import matplotlib.pylab as pl import ot +import ot.plot ############################################################################## # Generate data @@ -31,8 +32,8 @@ cov_s = np.array([[1, 0], [0, 1]]) mu_t = np.array([4, 4]) cov_t = np.array([[1, -.8], [-.8, 1]]) -xs = ot.datasets.get_2D_samples_gauss(n, mu_s, cov_s) -xt = ot.datasets.get_2D_samples_gauss(n, mu_t, cov_t) +xs = ot.datasets.make_2D_samples_gauss(n, mu_s, cov_s) +xt = ot.datasets.make_2D_samples_gauss(n, mu_t, cov_t) a, b = np.ones((n,)) / n, np.ones((n,)) / n # uniform distribution on samples |