From f6139428e70ce964de3bef703ef13aa701a83620 Mon Sep 17 00:00:00 2001 From: RĂ©mi Flamary Date: Tue, 22 Dec 2020 18:35:40 +0100 Subject: [WIP] Update documentation "Why OT" section (#220) * add some text + discussion sinkhorn * stating wrk on why POT * fix sphinx warnings + make html-noplot * discussion when not to use POT * add discussion which sinkhorn * edits on quickstart * more * remove warnings :any: * more * done * remove ref Co-authored-by: Alexandre Gramfort --- ot/__init__.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'ot/__init__.py') diff --git a/ot/__init__.py b/ot/__init__.py index ec3ede2..0116d33 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -37,7 +37,8 @@ from . import partial # 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, sinkhorn_unbalanced2 +from .unbalanced import (sinkhorn_unbalanced, barycenter_unbalanced, + sinkhorn_unbalanced2) from .da import sinkhorn_lpl1_mm from .sliced import sliced_wasserstein_distance @@ -46,9 +47,10 @@ from .utils import dist, unif, tic, toc, toq __version__ = "0.7.0" -__all__ = ['emd', 'emd2', 'emd_1d', 'sinkhorn', 'sinkhorn2', 'utils', 'datasets', - 'bregman', 'lp', 'tic', 'toc', 'toq', 'gromov', +__all__ = ['emd', 'emd2', 'emd_1d', 'sinkhorn', 'sinkhorn2', 'utils', + 'datasets', 'bregman', 'lp', 'tic', 'toc', 'toq', 'gromov', 'emd_1d', 'emd2_1d', 'wasserstein_1d', 'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim', 'sinkhorn_unbalanced', 'barycenter_unbalanced', - 'sinkhorn_unbalanced2', 'sliced_wasserstein_distance'] + 'sinkhorn_unbalanced2', 'sliced_wasserstein_distance', + 'smooth', 'stochastic', 'unbalanced', 'partial'] -- cgit v1.2.3