From 3547344740ec16157fd65c7f27d4acd6add986a4 Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Wed, 30 Aug 2017 17:10:38 +0200
Subject: exples +rst

---
 examples/plot_OT_1D.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'examples/plot_OT_1D.py')

diff --git a/examples/plot_OT_1D.py b/examples/plot_OT_1D.py
index 0f3a26a..be6f5b3 100644
--- a/examples/plot_OT_1D.py
+++ b/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
-- 
cgit v1.2.3