exples +rst

Author: rflamary

examples/plot_OT_1D.py

@@ -15,6 +15,10 @@
 import ot
 from ot.datasets import get_1D_gauss as gauss
 
+##############################################################################
+# Generate data
+##############################################################################
+
 #%% parameters
 
 n = 100  # nb bins
@@ -30,6 +34,11 @@
 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,13 +51,21 @@
 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)
 
 pl.figure(3, figsize=(5, 5))
 ot.plot.plot1D_mat(a, b, G0, 'OT matrix G0')
 
+##############################################################################
+# Solve Sinkhorn
+##############################################################################
+
 #%% Sinkhorn
 
 lambd = 1e-3

examples/plot_optim_OTreg.py

@@ -12,6 +12,10 @@
 import ot
 
 
+##############################################################################
+# Generate data 
+##############################################################################
+
 #%% parameters
 
 n = 100  # nb bins
@@ -27,13 +31,21 @@
 M = ot.dist(x.reshape((n, 1)), x.reshape((n, 1)))
 M /= M.max()
 
+##############################################################################
+# Solve EMD 
+##############################################################################
+
 #%% EMD
 
 G0 = ot.emd(a, b, M)
 
 pl.figure(3, figsize=(5, 5))
 ot.plot.plot1D_mat(a, b, G0, 'OT matrix G0')
 
+##############################################################################
+# Solve EMD with Frobenius norm regularization
+##############################################################################
+
 #%% Example with Frobenius norm regularization
 
 
@@ -52,6 +64,10 @@ def df(G):
 pl.figure(3)
 ot.plot.plot1D_mat(a, b, Gl2, 'OT matrix Frob. reg')
 
+##############################################################################
+# Solve EMD with entropic regularization
+##############################################################################
+
 #%% Example with entropic regularization
 
 
@@ -70,8 +86,11 @@ def df(G):
 pl.figure(4, figsize=(5, 5))
 ot.plot.plot1D_mat(a, b, Ge, 'OT matrix Entrop. reg')
 
-#%% Example with Frobenius norm + entropic regularization with gcg
+##############################################################################
+# Solve EMD with Frobenius norm + entropic regularization
+##############################################################################
 
+#%% Example with Frobenius norm + entropic regularization with gcg
 
 def f(G):
     return 0.5 * np.sum(G**2)