diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2021-11-04 11:00:09 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-11-04 11:00:09 +0100 |
| commit | 2fe69eb130827560ada704bc25998397c4357821 (patch) | |
| tree | 82973444cc4afc4c42cc7cdaf43a2ebd4b1a6a91 /ot/__init__.py | |
| parent | 9c6ac880d426b7577918b0c77bd74b3b01930ef6 (diff) | |
[MRG] Make gromov loss differentiable wrt matrices and weights (#302)
* grmov differentable
* new stuff
* test gromov gradients
* fgwdifferentiable
* fgw tested
* correc name test
* add awesome example with gromov optimizatrion
* pep8+ typos
* damn pep8
* thunbnail
* remove prints
Diffstat (limited to 'ot/__init__.py')
| -rw-r--r-- | ot/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index f20332c..4292b41 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -43,6 +43,8 @@ from .unbalanced import (sinkhorn_unbalanced, barycenter_unbalanced, sinkhorn_unbalanced2) from .da import sinkhorn_lpl1_mm from .sliced import sliced_wasserstein_distance, max_sliced_wasserstein_distance +from .gromov import (gromov_wasserstein, gromov_wasserstein2, + gromov_barycenters, fused_gromov_wasserstein, fused_gromov_wasserstein2) # utils functions from .utils import dist, unif, tic, toc, toq |