summaryrefslogtreecommitdiff
path: root/ot/__init__.py
diff options
context:
space:
mode:
authorAdrienCorenflos <adrien.corenflos@gmail.com>2020-10-22 09:28:53 +0100
committerGitHub <noreply@github.com>2020-10-22 10:28:53 +0200
commit78b44af2434f494c8f9e4c8c91003fbc0e1d4415 (patch)
tree013002f0a65918cee5eb95648965d4361f0c3dc2 /ot/__init__.py
parent7adc1b1aa73c55dc07983ff08dcb23fd71e9e8b6 (diff)
[MRG] Sliced wasserstein (#203)
* example for log treatment in bregman.py * Improve doc * Revert "example for log treatment in bregman.py" This reverts commit 9f51c14e * Add comments by Flamary * Delete repetitive description * Added raw string to avoid pbs with backslashes * Implements sliced wasserstein * Changed formatting of string for py3.5 support * Docstest, expected 0.0 and not 0. * Adressed comments by @rflamary * No 3d plot here * add sliced to the docs * Incorporate comments by @rflamary * add link to pdf Co-authored-by: RĂ©mi Flamary <remi.flamary@gmail.com>
Diffstat (limited to 'ot/__init__.py')
-rw-r--r--ot/__init__.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/__init__.py b/ot/__init__.py
index 0e6e2e2..ec3ede2 100644
--- a/ot/__init__.py
+++ b/ot/__init__.py
@@ -39,6 +39,7 @@ from .lp import emd, emd2, emd_1d, emd2_1d, wasserstein_1d
from .bregman import sinkhorn, sinkhorn2, barycenter
from .unbalanced import sinkhorn_unbalanced, barycenter_unbalanced, sinkhorn_unbalanced2
from .da import sinkhorn_lpl1_mm
+from .sliced import sliced_wasserstein_distance
# utils functions
from .utils import dist, unif, tic, toc, toq
@@ -50,4 +51,4 @@ __all__ = ['emd', 'emd2', 'emd_1d', 'sinkhorn', 'sinkhorn2', 'utils', 'datasets'
'emd_1d', 'emd2_1d', 'wasserstein_1d',
'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim',
'sinkhorn_unbalanced', 'barycenter_unbalanced',
- 'sinkhorn_unbalanced2']
+ 'sinkhorn_unbalanced2', 'sliced_wasserstein_distance']