diff options
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 5 |
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 |