diff options
Diffstat (limited to 'ot/bregman.py')
| -rw-r--r-- | ot/bregman.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index 5d93fd6..b749b13 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -39,6 +39,11 @@ def sinkhorn(a,b, M, reg,numItermax = 1000,stopThr=1e-9): loss matrix reg: float Regularization term >0 + numItermax: int, optional + Max number of iterations + stopThr: float, optional + Stop threshol on error (>0) + Returns |