summaryrefslogtreecommitdiff
path: root/ot/__init__.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2019-07-03 14:11:16 +0200
committerRémi Flamary <remi.flamary@gmail.com>2019-07-03 14:11:16 +0200
commit7402d344240ce94e33c53daff419d4356278d48f (patch)
treec5b1b8f10eaee4c2ecaa12f629255489c2481590 /ot/__init__.py
parentd3236cf0cab000b5604f8ede9ebcbdc19d8c213f (diff)
doc in modules
Diffstat (limited to 'ot/__init__.py')
-rw-r--r--ot/__init__.py29
1 files changed, 28 insertions, 1 deletions
diff --git a/ot/__init__.py b/ot/__init__.py
index 35d2ddd..35ae6fc 100644
--- a/ot/__init__.py
+++ b/ot/__init__.py
@@ -1,7 +1,33 @@
"""
This is the main module of the POT toolbox. It provides easy access to
-a number of functions described below.
+a number of sub-modules and functions described below.
+
+.. note::
+
+
+ Here is a list of the submodules and short description of what they contain.
+
+ - :any:`ot.lp` contains OT solvers for the exact (Linear Program) OT problems.
+ - :any:`ot.bregman` contains OT solvers for the entropic OT problems using
+ Bregman projections.
+ - :any:`ot.lp` contains OT solvers for the exact (Linear Program) OT problems.
+ - :any:`ot.smooth` contains OT solvers for the regularized (l2 and kl) smooth OT
+ problems.
+ - :any:`ot.gromov` contains solvers for Gromov-Wasserstein and Fused Gromov
+ Wasserstein problems.
+ - :any:`ot.optim` contains generic solvers OT based optimization problems
+ - :any:`ot.da` contains classes and function related to Monge mapping
+ estimation and Domain Adaptation (DA).
+ - :any:`ot.gpu` contains GPU (cupy) implementation of some OT solvers
+ - :any:`ot.dr` contains Dimension Reduction (DR) methods such as Wasserstein
+ Discriminant Analysis.
+ - :any:`ot.utils` contains utility functions such as distance computation and
+ timing.
+ - :any:`ot.datasets` contains toy dataset generation functions.
+ - :any:`ot.plot` contains visualization functions
+ - :any:`ot.stochastic` contains stochastic solvers for regularized OT.
+ - :any:`ot.unbalanced` contains solvers for regularized unbalanced OT.
.. warning::
The list of automatically imported sub-modules is as follows:
@@ -14,6 +40,7 @@ a number of functions described below.
- :any:`ot.dr` : depends on :code:`pymanopt` and :code:`autograd`.
- :any:`ot.gpu` : depends on :code:`cupy` and a CUDA GPU.
+ - :any:`ot.plot` : depends on :code:`matplotlib`
"""