summaryrefslogtreecommitdiff
path: root/README.md
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2017-07-07 08:51:45 +0200
committerRémi Flamary <remi.flamary@gmail.com>2017-07-07 08:51:45 +0200
commit47477c5782f87570de590ea423a082b71dd63241 (patch)
tree00575e528509242cdad2cd283d5e2e0c44e53349 /README.md
parent0fc1124e001932354ec3d229d198cba166cd0b0e (diff)
add sinkhorbn2 +v3
Diffstat (limited to 'README.md')
-rw-r--r--README.md5
1 files changed, 3 insertions, 2 deletions
diff --git a/README.md b/README.md
index 4aa4cc5..d53387b 100644
--- a/README.md
+++ b/README.md
@@ -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