diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-07 08:51:45 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-07 08:51:45 +0200 |
commit | 47477c5782f87570de590ea423a082b71dd63241 (patch) | |
tree | 00575e528509242cdad2cd283d5e2e0c44e53349 /README.md | |
parent | 0fc1124e001932354ec3d229d198cba166cd0b0e (diff) |
add sinkhorbn2 +v3
Diffstat (limited to 'README.md')
-rw-r--r-- | README.md | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -83,14 +83,15 @@ import ot # a,b are 1D histograms (sum to 1 and positive) # M is the ground cost matrix Wd=ot.emd2(a,b,M) # exact linear program +Wd_reg=ot.sinkhorn2(a,b,M,reg) # entropic regularized OT # if b is a matrix compute all distances to a and return a vector ``` * Compute OT matrix ```python # a,b are 1D histograms (sum to 1 and positive) # M is the ground cost matrix -Totp=ot.emd(a,b,M) # exact linear program -Totp_reg=ot.sinkhorn(a,b,M,reg) # entropic regularized OT +T=ot.emd(a,b,M) # exact linear program +T_reg=ot.sinkhorn(a,b,M,reg) # entropic regularized OT ``` * Compute Wasserstein barycenter ```python |