Skip to content

Commit 36f4f7e

Browse files
committed
better documentation
1 parent 4285cf6 commit 36f4f7e

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

ot/lp/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,12 @@
1111

1212
import numpy as np
1313

14+
from .import cvx
15+
1416
# import compiled emd
1517
from .emd_wrap import emd_c, check_result
1618
from ..utils import parmap
19+
from .cvx import barycenter
1720

1821

1922
def emd(a, b, M, numItermax=100000, log=False):

ot/lp/cvx.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ def barycenter(A, M, weights=None, verbose=False, log=False, solver='interior-po
3838
- :math:`W_1(\cdot,\cdot)` is the Wasserstein distance (see ot.emd.sinkhorn)
3939
- :math:`\mathbf{a}_i` are training distributions in the columns of matrix :math:`\mathbf{A}`
4040
41-
The linear program is solved using the default cvxopt solver if installed.
42-
If cvxopt is not installed it uses the lp solver from scipy.optimize.
41+
The linear program is solved using the interior point solver from scipy.optimize.
42+
If cvxopt solver if installed it can use cvxopt.
4343
4444
Parameters
4545
----------

0 commit comments

Comments
 (0)