From 40ee453736b792d4a3318ff2bc145d488c00d9a1 Mon Sep 17 00:00:00 2001 From: Oleksii Kachaiev Date: Wed, 23 Aug 2023 22:40:50 +0200 Subject: [PATCH] Use local rng for 2gauss_prop --- ot/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ot/datasets.py b/ot/datasets.py index 3d633f401..35e781618 100644 --- a/ot/datasets.py +++ b/ot/datasets.py @@ -155,7 +155,7 @@ def make_data_classif(dataset, n, nz=.5, theta=0, p=.5, random_state=None, **kwa elif dataset.lower() == '2gauss_prop': y = np.concatenate((np.ones(int(p * n)), np.zeros(int((1 - p) * n)))) - x = np.hstack((0 * y[:, None] - 0, 1 - 2 * y[:, None])) + nz * np.random.randn(len(y), 2) + x = np.hstack((0 * y[:, None] - 0, 1 - 2 * y[:, None])) + nz * generator.randn(len(y), 2) if ('bias' not in kwargs) and ('b' not in kwargs): kwargs['bias'] = np.array([0, 2])