summaryrefslogtreecommitdiff
path: root/ot/datasets.py
diff options
context:
space:
mode:
authorievred <ievgen.redko@univ-st-etienne.fr>2020-04-02 15:29:12 +0200
committerievred <ievgen.redko@univ-st-etienne.fr>2020-04-02 15:29:12 +0200
commit9200af5d795517b0772c10bb3d16022dd1a12791 (patch)
tree3220647499ea28b24c1126317700747e6c12c522 /ot/datasets.py
parent6b8477d1c08696a08a1b71642712d83e560f9623 (diff)
laplace v1
Diffstat (limited to 'ot/datasets.py')
-rw-r--r--ot/datasets.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/ot/datasets.py b/ot/datasets.py
index eea9f37..a1ca7b6 100644
--- a/ot/datasets.py
+++ b/ot/datasets.py
@@ -30,7 +30,7 @@ def make_1D_gauss(n, m, s):
1D histogram for a gaussian distribution
"""
x = np.arange(n, dtype=np.float64)
- h = np.exp(-(x - m)**2 / (2 * s**2))
+ h = np.exp(-(x - m) ** 2 / (2 * s ** 2))
return h / h.sum()
@@ -80,7 +80,7 @@ def get_2D_samples_gauss(n, m, sigma, random_state=None):
return make_2D_samples_gauss(n, m, sigma, random_state=None)
-def make_data_classif(dataset, n, nz=.5, theta=0, p = .5, random_state=None, **kwargs):
+def make_data_classif(dataset, n, nz=.5, theta=0, p=.5, random_state=None, **kwargs):
"""Dataset generation for classification problems
Parameters