diff options
author | ievred <ievgen.redko@univ-st-etienne.fr> | 2020-04-02 15:29:12 +0200 |
---|---|---|
committer | ievred <ievgen.redko@univ-st-etienne.fr> | 2020-04-02 15:29:12 +0200 |
commit | 9200af5d795517b0772c10bb3d16022dd1a12791 (patch) | |
tree | 3220647499ea28b24c1126317700747e6c12c522 /ot/datasets.py | |
parent | 6b8477d1c08696a08a1b71642712d83e560f9623 (diff) |
laplace v1
Diffstat (limited to 'ot/datasets.py')
-rw-r--r-- | ot/datasets.py | 4 |
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 |