summaryrefslogtreecommitdiff
path: root/examples/plot_barycenter_lp_vs_entropic.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/plot_barycenter_lp_vs_entropic.py')
-rw-r--r--examples/plot_barycenter_lp_vs_entropic.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/examples/plot_barycenter_lp_vs_entropic.py b/examples/plot_barycenter_lp_vs_entropic.py
index b82765e..d7c72d0 100644
--- a/examples/plot_barycenter_lp_vs_entropic.py
+++ b/examples/plot_barycenter_lp_vs_entropic.py
@@ -102,7 +102,7 @@ pl.tight_layout()
problems.append([A, [bary_l2, bary_wass, bary_wass2]])
##############################################################################
-# Dirac Data
+# Stair Data
# ----------
#%% parameters
@@ -168,6 +168,11 @@ pl.legend()
pl.title('Barycenters')
pl.tight_layout()
+
+##############################################################################
+# Dirac Data
+# ----------
+
#%% parameters
a1 = np.zeros(n)