summaryrefslogtreecommitdiff
path: root/ot/bregman.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2016-10-28 15:13:48 +0200
committerRémi Flamary <remi.flamary@gmail.com>2016-10-28 15:13:48 +0200
commit0d81de9909e8e9eb95858f0a043550b15898f172 (patch)
tree52a456562616b8d015c8985a072bcf8759e84fa7 /ot/bregman.py
parent4ced7428bd2be4ca008f12400afb445c5a6517c8 (diff)
doc da.py
Diffstat (limited to 'ot/bregman.py')
-rw-r--r--ot/bregman.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/ot/bregman.py b/ot/bregman.py
index ad9a67a..2d82ae4 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -6,12 +6,12 @@ Bregman projections for regularized OT
import numpy as np
-def sinkhorn(a,b, M, reg,numItermax = 1000,stopThr=1e-9,verbose=False,log=False):
+def sinkhorn(a,b, M, reg, numItermax = 1000, stopThr=1e-9, verbose=False, log=False):
"""
- Solve the entropic regularization optimal transport problem and return the OT matrix
-
+ Solve the entropic regularization optimal transport problem
+
The function solves the following optimization problem:
-
+
.. math::
\gamma = arg\min_\gamma <\gamma,M>_F + reg\cdot\Omega(\gamma)