diff options
author | Kilian Fatras <kilianfatras@dhcp-206-12-53-20.eduroam.wireless.ubc.ca> | 2018-08-28 17:24:07 -0700 |
---|---|---|
committer | Kilian Fatras <kilianfatras@dhcp-206-12-53-20.eduroam.wireless.ubc.ca> | 2018-08-28 17:24:07 -0700 |
commit | e885d78cc9608d791a9d1561d2f4e0b783ba0761 (patch) | |
tree | e03a553873f110d1b8e0f15cc6f9248c916a405c /docs/source/auto_examples/plot_OT_2D_samples.py | |
parent | 77b68901c5415ddc5d9ab5215a6fa97723de3de9 (diff) |
debug sgd dual
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 |