corrected typos in barycenter examples

Author: Kilian Fatras

examples/plot_barycenter_lp_vs_entropic.py

@@ -102,7 +102,7 @@
 problems.append([A, [bary_l2, bary_wass, bary_wass2]])
 
 ##############################################################################
-# Dirac Data
+# Stair Data
 # ----------
 
 #%% parameters
@@ -168,6 +168,11 @@
 pl.title('Barycenters')
 pl.tight_layout()
 
+
+##############################################################################
+# Dirac Data
+# ----------
+
 #%% parameters
 
 a1 = np.zeros(n)

examples/plot_free_support_barycenter.py

@@ -62,7 +62,7 @@
 pl.figure(1)
 for (x_i, b_i) in zip(measures_locations, measures_weights):
     color = np.random.randint(low=1, high=10 * N)
-    pl.scatter(x_i[:, 0], x_i[:, 1], s=b * 1000, label='input measure')
+    pl.scatter(x_i[:, 0], x_i[:, 1], s=b_i * 1000, label='input measure')
 pl.scatter(X[:, 0], X[:, 1], s=b * 1000, c='black', marker='^', label='2-Wasserstein barycenter')
 pl.title('Data measures and their barycenter')
 pl.legend(loc=0)