diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2016-11-04 10:10:44 +0100 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2016-11-04 10:10:44 +0100 |
| commit | 405f352dc562eb17a2d6d7ca17c2ce14b19f2668 (patch) | |
| tree | fff693f369f0e65eaa800a4b98a3c1da90db8f9e /ot/optim.py | |
| parent | 0ea30e9ca2398caeb853662012b09148a25cffff (diff) | |
add mapping estimation with kernels works!
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) |