diff options
Diffstat (limited to 'ot/optim.py')
| -rw-r--r-- | ot/optim.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ot/optim.py b/ot/optim.py index dcefd24..2b8f565 100644 --- a/ot/optim.py +++ b/ot/optim.py @@ -159,6 +159,8 @@ def cg(a,b,M,reg,f,df,G0=None,numItermax = 200,stopThr=1e-9,verbose=False,log=Fa # problem linearization Mi=M+reg*df(G) + # set M positive + Mi+=Mi.min() # solve linear program Gc=emd(a,b,Mi) |