diff options
author | Hicham Janati <hicham.janati@inria.fr> | 2019-09-03 18:15:25 +0200 |
---|---|---|
committer | Hicham Janati <hicham.janati@inria.fr> | 2019-09-03 18:15:25 +0200 |
commit | e55232056a79de128583b87e65abc6d7a75fb298 (patch) | |
tree | 3de24e46dc3f896691e46e602ac9d49bd0a6f513 /ot | |
parent | 49d9b5cf4eecefdc0fff4db6c43e85d16e478efb (diff) |
add unbalanced_barycenter import
Diffstat (limited to 'ot')
-rw-r--r-- | ot/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index 7d9615a..99f288e 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -65,7 +65,7 @@ from . import unbalanced # OT functions from .lp import emd, emd2, emd_1d, emd2_1d, wasserstein_1d from .bregman import sinkhorn, sinkhorn2, barycenter -from .unbalanced import sinkhorn_unbalanced, barycenter_unbalanced +from .unbalanced import sinkhorn_unbalanced, barycenter_unbalanced, sinkhorn_unbalanced2 from .da import sinkhorn_lpl1_mm # utils functions |