diff options
-rw-r--r-- | examples/demo_optim_OTreg.py | 9 | ||||
-rw-r--r-- | ot/optim.py | 1 |
2 files changed, 5 insertions, 5 deletions
diff --git a/examples/demo_optim_OTreg.py b/examples/demo_optim_OTreg.py index f12fdc1..9ffb02d 100644 --- a/examples/demo_optim_OTreg.py +++ b/examples/demo_optim_OTreg.py @@ -34,12 +34,13 @@ ot.plot.plot1D_mat(a,b,G0,'OT matrix G0') #%% exampel of regularization with Frobnisu norm def f(G): - return 0.5*np.sum(G**2) + #return 0.5*np.sum(G**2) + return np.sum(G*np.log(G)) def df(G): - return G - -reg=1e1 +# return G + return np.log(G)+1 +reg=1e-3 Greg=ot.optim.cg(a,b,M,reg,f,df,verbose=True) diff --git a/ot/optim.py b/ot/optim.py index 03ed83d..66bbfb5 100644 --- a/ot/optim.py +++ b/ot/optim.py @@ -122,7 +122,6 @@ def cg(a,b,M,reg,f,df,G0=None,numItermax = 200,stopThr=1e-9,verbose=False,log=Fa # line search alpha,fc,f_val = line_search_armijo(cost,G,deltaG,Mi,f_val) - G=G+alpha*deltaG # test convergence |