Skip to content

Commit cfdbbd2

Browse files
author
Hicham Janati
committed
remove square in convergence check
add unbalanced with stabilization add unbalanced tests with stabilization fix doctest examples add xvfb in travis remove explicit call xvfb in travis change alpha to reg_m minor flake8 remove redundant sink definitions + better doc and naming add stabilized unbalanced barycenter + add not converged warnings add test for stable barycenter add generic barycenter func + make method funcs private fix typo + add method test for barycenters fix doc examples + add xml to gitignore fix whitespace in example change logsumexp import - scipy deprecation warning fix doctest improve naming + add stable barycenter in bregman add test for stable bar + test the method arg in bregman
1 parent abfe183 commit cfdbbd2

7 files changed

+1205
-384
lines changed

.gitignore

+3
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,9 @@ coverage.xml
5959
*.mo
6060
*.pot
6161

62+
# xml
63+
*.xml
64+
6265
# Django stuff:
6366
*.log
6467
local_settings.py

ot/__init__.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""
22
3-
This is the main module of the POT toolbox. It provides easy access to
4-
a number of sub-modules and functions described below.
3+
This is the main module of the POT toolbox. It provides easy access to
4+
a number of sub-modules and functions described below.
55
66
.. note::
77
@@ -14,27 +14,27 @@
1414
- :any:`ot.lp` contains OT solvers for the exact (Linear Program) OT problems.
1515
- :any:`ot.smooth` contains OT solvers for the regularized (l2 and kl) smooth OT
1616
problems.
17-
- :any:`ot.gromov` contains solvers for Gromov-Wasserstein and Fused Gromov
17+
- :any:`ot.gromov` contains solvers for Gromov-Wasserstein and Fused Gromov
1818
Wasserstein problems.
19-
- :any:`ot.optim` contains generic solvers OT based optimization problems
19+
- :any:`ot.optim` contains generic solvers OT based optimization problems
2020
- :any:`ot.da` contains classes and function related to Monge mapping
2121
estimation and Domain Adaptation (DA).
2222
- :any:`ot.gpu` contains GPU (cupy) implementation of some OT solvers
23-
- :any:`ot.dr` contains Dimension Reduction (DR) methods such as Wasserstein
23+
- :any:`ot.dr` contains Dimension Reduction (DR) methods such as Wasserstein
2424
Discriminant Analysis.
25-
- :any:`ot.utils` contains utility functions such as distance computation and
26-
timing.
25+
- :any:`ot.utils` contains utility functions such as distance computation and
26+
timing.
2727
- :any:`ot.datasets` contains toy dataset generation functions.
2828
- :any:`ot.plot` contains visualization functions
2929
- :any:`ot.stochastic` contains stochastic solvers for regularized OT.
3030
- :any:`ot.unbalanced` contains solvers for regularized unbalanced OT.
3131
3232
.. warning::
33-
The list of automatically imported sub-modules is as follows:
33+
The list of automatically imported sub-modules is as follows:
3434
:py:mod:`ot.lp`, :py:mod:`ot.bregman`, :py:mod:`ot.optim`
3535
:py:mod:`ot.utils`, :py:mod:`ot.datasets`,
3636
:py:mod:`ot.gromov`, :py:mod:`ot.smooth`
37-
:py:mod:`ot.stochastic`
37+
:py:mod:`ot.stochastic`
3838
3939
The following sub-modules are not imported due to additional dependencies:
4040

0 commit comments

Comments
 (0)