diff options
| author | haoran010 <62598274+haoran010@users.noreply.github.com> | 2021-10-25 10:47:22 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-10-25 10:47:22 +0200 |
| commit | 7af8c2147d61349f4d99ca33318a8a125e4569aa (patch) | |
| tree | 5c08a89f2c998a6c1d734be28e4127130c1c1102 /ot/__init__.py | |
| parent | d50d8145a5c0cf69d438b018cd5f1b914905e784 (diff) | |
[MRG] Regularization path for l2 UOT (#274)
* add reg path
* debug examples and verify pep8
* pep8 and move the reg path examples in unbalanced folder
Co-authored-by: haoran010 <haoran.wu@insa-rennes.fr>
Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
Diffstat (limited to 'ot/__init__.py')
| -rw-r--r-- | ot/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index 3b072c6..5bd4bab 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -34,6 +34,7 @@ from . import stochastic from . import unbalanced from . import partial from . import backend +from . import regpath # OT functions from .lp import emd, emd2, emd_1d, emd2_1d, wasserstein_1d @@ -54,4 +55,4 @@ __all__ = ['emd', 'emd2', 'emd_1d', 'sinkhorn', 'sinkhorn2', 'utils', 'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim', 'sinkhorn_unbalanced', 'barycenter_unbalanced', 'sinkhorn_unbalanced2', 'sliced_wasserstein_distance', - 'smooth', 'stochastic', 'unbalanced', 'partial'] + 'smooth', 'stochastic', 'unbalanced', 'partial', 'regpath'] |