diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2018-06-11 11:24:57 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-06-11 11:24:57 +0200 |
commit | 47730fc727c0f54e8459964d9208ad824e3f91da (patch) | |
tree | 48952166e88e602f1843bd15c0187c7d5ffb6cac /docs/source/auto_examples/plot_OT_2D_samples.py | |
parent | 4641ec5f2ddbff1a468afaf65741aecae44738cc (diff) | |
parent | 530dc93a60e9b81fb8d1b44680deea77dacf660b (diff) |
Merge branch 'master' into remove_otda_v05
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 |