summaryrefslogtreecommitdiff
path: root/ot/optim.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2016-10-26 16:49:52 +0200
committerRémi Flamary <remi.flamary@gmail.com>2016-10-26 16:49:52 +0200
commitb8d5e6baac6b57e99eeff7f67a76700aa04e1b24 (patch)
tree1f51ca33a3107df2635b0651a4afc13b02290480 /ot/optim.py
parent22dcef6c1cf6bd7ae626ef768359c7495298ba29 (diff)
add conditionnal gradient and demo
Diffstat (limited to 'ot/optim.py')
-rw-r--r--ot/optim.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/optim.py b/ot/optim.py
index 86371b9..03ed83d 100644
--- a/ot/optim.py
+++ b/ot/optim.py
@@ -120,7 +120,7 @@ def cg(a,b,M,reg,f,df,G0=None,numItermax = 200,stopThr=1e-9,verbose=False,log=Fa
deltaG=Gc-G
# line search
- alpha,fc,f_val = line_search_armijo(f,G,deltaG,Mi,f_val)
+ alpha,fc,f_val = line_search_armijo(cost,G,deltaG,Mi,f_val)
G=G+alpha*deltaG