diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-20 15:19:09 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-20 15:19:09 +0200 |
commit | e65606ae498bd611f6a994868c2a66dfbea403cd (patch) | |
tree | b9b43dcaf8499b6d57b806ce04350fb6b792537f /docs/source/auto_examples/plot_UOT_barycenter_1D.py | |
parent | 8acaf262baa04a4d2bdd9c774c45c5bb2fb2d12a (diff) |
big update examples
Diffstat (limited to 'docs/source/auto_examples/plot_UOT_barycenter_1D.py')
-rw-r--r-- | docs/source/auto_examples/plot_UOT_barycenter_1D.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/docs/source/auto_examples/plot_UOT_barycenter_1D.py b/docs/source/auto_examples/plot_UOT_barycenter_1D.py index c8d9d3b..acb5892 100644 --- a/docs/source/auto_examples/plot_UOT_barycenter_1D.py +++ b/docs/source/auto_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 |