From 830d4ebd2e2c85b4f3503f358bb31a07918a27c5 Mon Sep 17 00:00:00 2001 From: Rémi Flamary Date: Tue, 25 Jun 2019 14:27:19 +0200 Subject: update fgw example + add notebook --- examples/plot_barycenter_fgw.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'examples/plot_barycenter_fgw.py') 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 -- cgit v1.2.3