diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2017-06-20 14:51:12 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-06-20 14:51:12 +0200 |
| commit | 77bcf836272faf1a40fdea97131b85390e935be3 (patch) | |
| tree | 4ab08952b9f67183f30d6a01af59c654075aaeb7 /ot/bregman.py | |
| parent | 2bcc24aa05078cfbf160be06fc9ad166bee52904 (diff) | |
add clean zeros function for sparse distributions
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 |