Skip to content

Commit 830d4eb

Browse files
committed
update fgw example + add notebook
1 parent c4b0aeb commit 830d4eb

File tree

4 files changed

+756
-8
lines changed

4 files changed

+756
-8
lines changed

examples/plot_barycenter_fgw.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ def graph_colors(nx_graph, vmin=0, vmax=7):
166166
lambdas = np.array([np.ones(len(Ys)) / len(Ys)]).ravel()
167167
sizebary = 15 # we choose a barycenter with 15 nodes
168168

169-
A, C, log = fgw_barycenters(sizebary, Ys, Cs, ps, lambdas, alpha=0.95)
169+
A, C, log = fgw_barycenters(sizebary, Ys, Cs, ps, lambdas, alpha=0.95, log=True)
170170

171171
##############################################################################
172172
# Plot Barycenter

0 commit comments

Comments
 (0)