summaryrefslogtreecommitdiff
path: root/ot/gromov.py
diff options
context:
space:
mode:
authorRĂ©mi Flamary <remi.flamary@gmail.com>2019-03-21 14:44:56 +0100
committerGitHub <noreply@github.com>2019-03-21 14:44:56 +0100
commit2384380536e3cc405e4db9f4b31cb48d309f257c (patch)
tree21ef9e521925e82fa1f220a3e12f6d7be21dd501 /ot/gromov.py
parente57f41a9f6942ef2d18fde214b90c5dee57e5c8d (diff)
parentb7df4fbbb54eba6a797c6b8f0d57bda7db8e4177 (diff)
Merge pull request #79 from rflamary/update_doc
Update doc + notebooks
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 0278e99..7974546 100644
--- a/ot/gromov.py
+++ b/ot/gromov.py
@@ -39,7 +39,7 @@ def init_matrix(C1, C2, T, 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