Skip to content
This repository was archived by the owner on Dec 6, 2023. It is now read-only.

Commit 34967a7

Browse files
authored
Update import of make_classification() and make_regression() functions (#162)
* Update import of `make_classification()` * Update test_dataset.py * Update test_dual_cd.py * Update test_sgd.py
1 parent c4b315d commit 34967a7

File tree

4 files changed

+3
-4
lines changed

4 files changed

+3
-4
lines changed

examples/trace.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from scipy.linalg import svd
1010

1111
from sklearn.datasets import fetch_20newsgroups_vectorized
12-
from sklearn.datasets.samples_generator import make_classification
1312
from sklearn.feature_selection import SelectKBest, chi2
1413

1514
from lightning.classification import FistaClassifier

lightning/impl/tests/test_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from sklearn.utils.testing import assert_equal
88
from six.moves import xrange
99

10-
from sklearn.datasets.samples_generator import make_classification
10+
from sklearn.datasets import make_classification
1111
from sklearn.utils import check_random_state
1212

1313
from lightning.impl.dataset_fast import ContiguousDataset

lightning/impl/tests/test_dual_cd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import scipy.sparse as sp
33

44
from sklearn.metrics.pairwise import linear_kernel
5-
from sklearn.datasets.samples_generator import make_regression
5+
from sklearn.datasets import make_regression
66
from six.moves import xrange
77

88
from sklearn.utils.testing import assert_equal

lightning/impl/tests/test_sgd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from sklearn.utils.testing import assert_greater
77
from sklearn.utils.testing import assert_equal
88

9-
from sklearn.datasets.samples_generator import make_regression
9+
from sklearn.datasets import make_regression
1010

1111
from lightning.impl.datasets.samples_generator import make_classification
1212
from lightning.impl.datasets.samples_generator import make_nn_regression

0 commit comments

Comments
 (0)