demo+sinkhorn

1 files changed, 10 insertions, 0 deletions

diff --git a/ot/datasets.py b/ot/datasets.py
new file mode 100644
index 0000000..bb10ba4
--- /dev/null
+++ b/ot/datasets.py
@@ -0,0 +1,10 @@
+
+import numpy as np
+
+
+
+def get_1D_gauss(n,m,s):
+    "return a 1D histogram for a gaussian distribution (n bins, mean m and std s) "
+    x=np.arange(n,dtype=np.float64)
+    h=np.exp(-(x-m)**2/(2*s^2))
+    return h/h.sum()
\ No newline at end of file