Skip to content

Commit e39476b

Browse files
committed
it should work now
1 parent 71ad677 commit e39476b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ot/gromov/_gw.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -586,6 +586,7 @@ def fused_gromov_wasserstein2(M, C1, C2, p=None, q=None, loss_fun='square_loss',
586586
lin_term = nx.sum(T * M)
587587
log_fgw['quad_loss'] = (fgw_dist - (1 - alpha) * lin_term)
588588
log_fgw['lin_loss'] = lin_term * (1 - alpha)
589+
gw_term = log_fgw['quad_loss'] / alpha
589590

590591
if loss_fun == 'square_loss':
591592
gC1 = 2 * C1 * nx.outer(p, p) - 2 * nx.dot(T, nx.dot(C2, T.T))

0 commit comments

Comments
 (0)