diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2019-06-25 14:27:19 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2019-06-25 14:27:19 +0200 |
commit | 830d4ebd2e2c85b4f3503f358bb31a07918a27c5 (patch) | |
tree | 7d91d64761882ba3d4fb4f08aaf37a32907fec10 /examples | |
parent | c4b0aeb20d920ba366a656a9aee7afe78871c9c7 (diff) |
update fgw example + add notebook
Diffstat (limited to 'examples')
-rw-r--r-- | examples/plot_barycenter_fgw.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/plot_barycenter_fgw.py b/examples/plot_barycenter_fgw.py index e4be447..77b0370 100644 --- a/examples/plot_barycenter_fgw.py +++ b/examples/plot_barycenter_fgw.py @@ -166,7 +166,7 @@ Ys = [np.array([v for (k, v) in nx.get_node_attributes(x, 'attr_name').items()]) lambdas = np.array([np.ones(len(Ys)) / len(Ys)]).ravel() sizebary = 15 # we choose a barycenter with 15 nodes -A, C, log = fgw_barycenters(sizebary, Ys, Cs, ps, lambdas, alpha=0.95) +A, C, log = fgw_barycenters(sizebary, Ys, Cs, ps, lambdas, alpha=0.95, log=True) ############################################################################## # Plot Barycenter |