diff options
| author | tvayer <titouan.vayer@gmail.com> | 2019-05-29 15:10:37 +0200 |
|---|---|---|
| committer | tvayer <titouan.vayer@gmail.com> | 2019-05-29 15:10:37 +0200 |
| commit | 103dfe0ee76e110bb9e0d1e36e3dd86109db3fce (patch) | |
| tree | 7d905c487a2950f69bffdc509244fcc145f2baca /ot/gromov.py | |
| parent | fa989062c17f87bd96aa58ad764fd3791ea11e22 (diff) | |
test check
Diffstat (limited to 'ot/gromov.py')
| -rw-r--r-- | ot/gromov.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/gromov.py b/ot/gromov.py index 33134a2..44248d1 100644 --- a/ot/gromov.py +++ b/ot/gromov.py @@ -348,7 +348,7 @@ def gromov_wasserstein(C1, C2, p, q, loss_fun, log=False, amijo=False, **kwargs) log['gw_dist'] = gwloss(constC, hC1, hC2, res)
return res, log
else:
- return cg(p, q, 0, 1, f, df, G0, amijo=amijo, **kwargs)
+ return cg(p, q, 0, 1, f, df, G0, amijo=amijo, C1=C1, C2=C2, constC=constC, **kwargs)
def fused_gromov_wasserstein(M, C1, C2, p, q, loss_fun='square_loss', alpha=0.5, amijo=False, **kwargs):
|