From 242c10d93e17eed006307d8f1fa0af8733cf3307 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20Flamary?= Date: Wed, 18 Oct 2023 15:41:32 +0200 Subject: [PATCH] small bug entropic fgw --- ot/gromov/_bregman.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ot/gromov/_bregman.py b/ot/gromov/_bregman.py index 654daf232..ddf34f793 100644 --- a/ot/gromov/_bregman.py +++ b/ot/gromov/_bregman.py @@ -826,10 +826,10 @@ def entropic_fused_gromov_wasserstein2( logv['T'] = T lin_term = nx.sum(T * M) - gw_term = (logv['gw_dist'] - (1 - alpha) * lin_term) / alpha + gw_term = (logv['fgw_dist'] - (1 - alpha) * lin_term) / alpha - log_fgw['quad_loss'] = gw_term * alpha - log_fgw['lin_loss'] = lin_term * (1 - alpha) + logv['quad_loss'] = gw_term * alpha + logv['lin_loss'] = lin_term * (1 - alpha) if log: return logv['fgw_dist'], logv