diff options
Diffstat (limited to 'ot/bregman.py')
| -rw-r--r-- | ot/bregman.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index e847f24..a13345d 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -108,6 +108,8 @@ def sinkhorn(a,b, M, reg,method='sinkhorn', numItermax = 1000, stopThr=1e-9, ver return sink() + + def sinkhorn_knopp(a,b, M, reg, numItermax = 1000, stopThr=1e-9, verbose=False, log=False,**kwargs): """ Solve the entropic regularization optimal transport problem and return the OT matrix |