diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-20 12:41:42 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-04-20 12:41:42 +0200 |
| commit | df293135613fba35d525b27751f847abbef64f88 (patch) | |
| tree | cb903103c8b58d3bc76ee3178bfc63596c3542cc /ot/__init__.py | |
| parent | adc5570550676b63b9aabb2205a67c5b7c9187f3 (diff) | |
| parent | de5679ad6e284c7d8751bcef05d4be5fd79d0aec (diff) | |
Merge pull request #141 from lchapel/partial-W-and-GW
[MRG] partial Wass and GW
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 89c7936..4fcb800 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -28,6 +28,7 @@ a number of sub-modules and functions described below. - :any:`ot.plot` contains visualization functions - :any:`ot.stochastic` contains stochastic solvers for regularized OT. - :any:`ot.unbalanced` contains solvers for regularized unbalanced OT. + - :any:`ot.partial` contains solvers for partial OT. .. warning:: The list of automatically imported sub-modules is as follows: @@ -61,6 +62,7 @@ from . import gromov from . import smooth from . import stochastic from . import unbalanced +from . import partial # OT functions from .lp import emd, emd2, emd_1d, emd2_1d, wasserstein_1d |