summaryrefslogtreecommitdiff
path: root/ot/gromov.py
diff options
context:
space:
mode:
authortvayer <titouan.vayer@gmail.com>2019-05-29 14:24:05 +0200
committertvayer <titouan.vayer@gmail.com>2019-05-29 14:24:05 +0200
commit63bbeb34e48f02c97a762dab5232158d90a5cffc (patch)
tree853026b5854b6e4b01fdf750db139985b3dd596f /ot/gromov.py
parentf70aabfcc11f92181e0dc987b341bad8ec030d75 (diff)
parentf66ab58c7c895011fd37bafd3e848828399c56c4 (diff)
Merge remote-tracking branch 'rflamary/master'
merge pot
Diffstat (limited to 'ot/gromov.py')
-rw-r--r--ot/gromov.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/gromov.py b/ot/gromov.py
index fe4fc15..33134a2 100644
--- a/ot/gromov.py
+++ b/ot/gromov.py
@@ -40,7 +40,7 @@ def init_matrix(C1, C2, p, q, loss_fun='square_loss'):
* h1(a)=a
* h2(b)=b
- The kl-loss function L(a,b)=(1/2)*|a-b|^2 is read as :
+ The kl-loss function L(a,b)=a*log(a/b)-a+b is read as :
L(a,b) = f1(a)+f2(b)-h1(a)*h2(b) with :
* f1(a)=a*log(a)-a
* f2(b)=b