diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2022-03-24 14:13:25 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-03-24 14:13:25 +0100 |
| commit | 82452e0f5f6dae05c7a1cc384e7a1fb62ae7e0d5 (patch) | |
| tree | 051871e3dc63e6bba1d0ecb1df6229796edd33bb /ot/__init__.py | |
| parent | 767171593f2a98a26b9a39bf110a45085e3b982e (diff) | |
[MRG] Add factored coupling (#358)
* add gfactored ot
* pep8 and add doc
* add exmaple for factotred OT
* final number of PR
* correct test on backends
* remove useless loss
* better tests
Diffstat (limited to 'ot/__init__.py')
| -rw-r--r-- | ot/__init__.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index bda7a35..c5e1967 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -33,6 +33,7 @@ from . import partial from . import backend from . import regpath from . import weak +from . import factored # OT functions from .lp import emd, emd2, emd_1d, emd2_1d, wasserstein_1d @@ -44,6 +45,9 @@ 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) from .weak import weak_optimal_transport +from .factored import factored_optimal_transport + + # utils functions from .utils import dist, unif, tic, toc, toq @@ -57,4 +61,5 @@ __all__ = ['emd', 'emd2', 'emd_1d', 'sinkhorn', 'sinkhorn2', 'utils', 'sinkhorn_unbalanced2', 'sliced_wasserstein_distance', 'gromov_wasserstein', 'gromov_wasserstein2', 'gromov_barycenters', 'fused_gromov_wasserstein', 'fused_gromov_wasserstein2', 'max_sliced_wasserstein_distance', 'weak_optimal_transport', + 'factored_optimal_transport', 'smooth', 'stochastic', 'unbalanced', 'partial', 'regpath'] |