diff options
Diffstat (limited to 'examples/plot_barycenter_fgw.py')
-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 |