summaryrefslogtreecommitdiff
path: root/examples
diff options
context:
space:
mode:
authorRĂ©mi Flamary <remi.flamary@gmail.com>2019-07-03 14:34:13 +0200
committerGitHub <noreply@github.com>2019-07-03 14:34:13 +0200
commit952503e02b1fc9bdf0811b937baacca57e4a98f1 (patch)
treec5b1b8f10eaee4c2ecaa12f629255489c2481590 /examples
parent8b3927bb5e8935c3dbddf054f054dc0c036fbdfe (diff)
parent7402d344240ce94e33c53daff419d4356278d48f (diff)
Merge pull request #88 from rflamary/doc_modules
[MRG] Update documentation and add quick start guide
Diffstat (limited to 'examples')
-rw-r--r--examples/plot_barycenter_fgw.py2
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