diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-24 14:18:41 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2020-04-24 14:18:41 +0200 |
| commit | 17d388be57cb5b0b2492c6b0ad8940e58b36016a (patch) | |
| tree | f7d0cb9d67cf716a043c6bd82b283695048af12c /ot | |
| parent | 90bd408e86eccb03b02d57a0cd7963e0c848a1fc (diff) | |
test raise un partial ot
Diffstat (limited to 'ot')
| -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 a1ca7b6..daca1ae 100644 --- a/ot/datasets.py +++ b/ot/datasets.py @@ -147,8 +147,8 @@ def make_data_classif(dataset, n, nz=.5, theta=0, p=.5, random_state=None, **kwa n2 = np.sum(y == 2) x = np.zeros((n, 2)) - x[y == 1, :] = get_2D_samples_gauss(n1, m1, nz, random_state=generator) - x[y == 2, :] = get_2D_samples_gauss(n2, m2, nz, random_state=generator) + x[y == 1, :] = make_2D_samples_gauss(n1, m1, nz, random_state=generator) + x[y == 2, :] = make_2D_samples_gauss(n2, m2, nz, random_state=generator) x = x.dot(rot) |