From 581c6de782dca279edd97778cc474e7597788c0f Mon Sep 17 00:00:00 2001 From: Rémi Flamary Date: Fri, 21 Oct 2016 10:51:27 +0200 Subject: demo+sinkhorn --- ot/datasets.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 ot/datasets.py (limited to 'ot/datasets.py') 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 -- cgit v1.2.3