Skip to content

Commit 7da2d99

Browse files
committed
change example for working (default parameter change)
1 parent d5efa20 commit 7da2d99

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/plot_UOT_barycenter_1D.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@
7777
reg = 1e-3
7878
alpha = 1.
7979

80-
bary_wass = ot.unbalanced.barycenter_unbalanced(A, M, reg, alpha, weights)
80+
bary_wass = ot.unbalanced.barycenter_unbalanced(A, M, reg, alpha, weights=weights)
8181

8282
pl.figure(2)
8383
pl.clf()
@@ -111,7 +111,7 @@
111111
weight = weight_list[i]
112112
weights = np.array([1 - weight, weight])
113113
B_l2[:, i] = A.dot(weights)
114-
B_wass[:, i] = ot.unbalanced.barycenter_unbalanced(A, M, reg, alpha, weights)
114+
B_wass[:, i] = ot.unbalanced.barycenter_unbalanced(A, M, reg, alpha, weights=weights)
115115

116116

117117
# plot interpolation

0 commit comments

Comments
 (0)