diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-20 14:54:30 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-20 14:54:30 +0200 |
| commit | 7da2d99d628f71b08cefb01fc31d57d76196bcb7 (patch) | |
| tree | d3f1759b693328c54eb37a75951d4a6f3dfd10ab /examples | |
| parent | d5efa20c70a6988a251acc15759dcd9d182c6dc1 (diff) | |
change example for working (default parameter change)
Diffstat (limited to 'examples')
| -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 |