summaryrefslogtreecommitdiff
path: root/ot/bregman.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2016-10-28 08:53:16 +0200
committerRémi Flamary <remi.flamary@gmail.com>2016-10-28 08:53:16 +0200
commitcf2d92e151f816e6ddcfc4b64cbda1f8f7bde9df (patch)
tree11cc4cc57453a1ce1739604bd6d64bb6a3bf67e8 /ot/bregman.py
parent123a7c70488d68bff85bbad50c6dd8d390f1e728 (diff)
complete doc emd
Diffstat (limited to 'ot/bregman.py')
-rw-r--r--ot/bregman.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/ot/bregman.py b/ot/bregman.py
index 17ec06f..0d2c099 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -37,7 +37,7 @@ def sinkhorn(a,b, M, reg,numItermax = 1000,stopThr=1e-9):
samples in the target domain
M : np.ndarray (ns,nt)
loss matrix
- reg: float()
+ reg: float
Regularization term >0
@@ -54,7 +54,8 @@ def sinkhorn(a,b, M, reg,numItermax = 1000,stopThr=1e-9):
See Also
--------
- ot.emd.emd : Unregularized optimal ransport
+ ot.lp.emd : Unregularized OT
+ ot.optim.cg : General regularized OT
"""
# init data