summaryrefslogtreecommitdiff
path: root/ot/lp/__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/lp/__init__.py
parentd3236cf0cab000b5604f8ede9ebcbdc19d8c213f (diff)
doc in modules
Diffstat (limited to 'ot/lp/__init__.py')
-rw-r--r--ot/lp/__init__.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/ot/lp/__init__.py b/ot/lp/__init__.py
index aed29f8..17f1731 100644
--- a/ot/lp/__init__.py
+++ b/ot/lp/__init__.py
@@ -544,14 +544,13 @@ def wasserstein_1d(x_a, x_b, a=None, b=None, p=1.):
r"""Solves the p-Wasserstein distance problem between 1d measures and returns
the distance
-
.. math::
- \gamma = arg\min_\gamma \left( \sum_i \sum_j \gamma_{ij}
- |x_a[i] - x_b[j]|^p \\right)^{1/p}
+ \min_\gamma \left( \sum_i \sum_j \gamma_{ij} \|x_a[i] - x_b[j]\|^p \right)^{1/p}
s.t. \gamma 1 = a,
\gamma^T 1= b,
\gamma\geq 0
+
where :
- x_a and x_b are the samples