diff options
-rw-r--r-- | examples/plot_UOT_barycenter_1D.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/plot_UOT_barycenter_1D.py b/examples/plot_UOT_barycenter_1D.py index c8d9d3b..acb5892 100644 --- a/examples/plot_UOT_barycenter_1D.py +++ b/examples/plot_UOT_barycenter_1D.py @@ -77,7 +77,7 @@ bary_l2 = A.dot(weights) reg = 1e-3 alpha = 1. -bary_wass = ot.unbalanced.barycenter_unbalanced(A, M, reg, alpha, weights) +bary_wass = ot.unbalanced.barycenter_unbalanced(A, M, reg, alpha, weights=weights) pl.figure(2) pl.clf() @@ -111,7 +111,7 @@ for i in range(0, n_weight): weight = weight_list[i] weights = np.array([1 - weight, weight]) B_l2[:, i] = A.dot(weights) - B_wass[:, i] = ot.unbalanced.barycenter_unbalanced(A, M, reg, alpha, weights) + B_wass[:, i] = ot.unbalanced.barycenter_unbalanced(A, M, reg, alpha, weights=weights) # plot interpolation |