summaryrefslogtreecommitdiff
path: root/ot/bregman
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2016-10-21 11:19:46 +0200
committerRémi Flamary <remi.flamary@gmail.com>2016-10-21 11:19:46 +0200
commit872e6db7c0d110069b450cbe7efcc186c4871428 (patch)
treef499a0258cf69f47a211a54447af990ac0afb591 /ot/bregman
parent581c6de782dca279edd97778cc474e7597788c0f (diff)
demo with sinkhorn
Diffstat (limited to 'ot/bregman')
-rw-r--r--ot/bregman/sink.py (renamed from ot/bregman/sinkhorn.py)2
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/bregman/sinkhorn.py b/ot/bregman/sink.py
index 798ac97..8b97e1e 100644
--- a/ot/bregman/sinkhorn.py
+++ b/ot/bregman/sink.py
@@ -59,7 +59,7 @@ def sinkhorn(a,b, M, reg,numItermax = 1000,stopThr=1e-9):
#print reg
- K = np.exp(-reg*M)
+ K = np.exp(-M/reg)
#print np.min(K)
Kp = np.dot(np.diag(1/a),K)