diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/import_data.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/import_data.py b/test/import_data.py index 6d4bd9e..158e97d 100644 --- a/test/import_data.py +++ b/test/import_data.py @@ -37,7 +37,7 @@ def uniform_data(n_observations = 1000, n_features = 50, n_relevant = 5): import numpy as np xmax = 10 xmin = 0 - data = np.random.randint(xmax + 1, size = (n_features, n_observations)) + data = 1.0*np.random.randint(xmax + 1, size = (n_features, n_observations)) labels = np.zeros(n_observations) delta = n_relevant * (xmax - xmin) / 2.0 |