diff options
Diffstat (limited to 'docs/source/auto_examples/plot_OT_1D.py')
| -rw-r--r-- | docs/source/auto_examples/plot_OT_1D.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/docs/source/auto_examples/plot_OT_1D.py b/docs/source/auto_examples/plot_OT_1D.py index 0f3a26a..be6f5b3 100644 --- a/docs/source/auto_examples/plot_OT_1D.py +++ b/docs/source/auto_examples/plot_OT_1D.py @@ -15,6 +15,10 @@ import matplotlib.pylab as pl import ot from ot.datasets import get_1D_gauss as gauss +############################################################################## +# Generate data +############################################################################## + #%% parameters n = 100 # nb bins @@ -30,6 +34,11 @@ b = gauss(n, m=60, s=10) M = ot.dist(x.reshape((n, 1)), x.reshape((n, 1))) M /= M.max() + +############################################################################## +# Plot distributions and loss matrix +############################################################################## + #%% plot the distributions pl.figure(1, figsize=(6.4, 3)) @@ -42,6 +51,10 @@ pl.legend() pl.figure(2, figsize=(5, 5)) ot.plot.plot1D_mat(a, b, M, 'Cost matrix M') +############################################################################## +# Solve EMD +############################################################################## + #%% EMD G0 = ot.emd(a, b, M) @@ -49,6 +62,10 @@ G0 = ot.emd(a, b, M) pl.figure(3, figsize=(5, 5)) ot.plot.plot1D_mat(a, b, G0, 'OT matrix G0') +############################################################################## +# Solve Sinkhorn +############################################################################## + #%% Sinkhorn lambd = 1e-3 |