diff options
Diffstat (limited to 'examples/plot_OT_1D.py')
| -rw-r--r-- | examples/plot_OT_1D.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/examples/plot_OT_1D.py b/examples/plot_OT_1D.py index b6ffa5f..719058f 100644 --- a/examples/plot_OT_1D.py +++ b/examples/plot_OT_1D.py @@ -20,7 +20,7 @@ from ot.datasets import get_1D_gauss as gauss ############################################################################## # Generate data -# ############# +# ------------- #%% parameters @@ -41,7 +41,7 @@ M /= M.max() ############################################################################## # Plot distributions and loss matrix -################################### +# ---------------------------------- #%% plot the distributions @@ -57,7 +57,8 @@ ot.plot.plot1D_mat(a, b, M, 'Cost matrix M') ############################################################################## # Solve EMD -############################################################################## +# --------- + #%% EMD @@ -68,7 +69,8 @@ ot.plot.plot1D_mat(a, b, G0, 'OT matrix G0') ############################################################################## # Solve Sinkhorn -############################################################################## +# -------------- + #%% Sinkhorn |