Skip to content

Commit 2e29414

Browse files
authored
Merge pull request rasbt#1037 from namanmistry/load_boston_fixed_test
Fixed load_boston test_exhaustive_feature_selector
2 parents aaa207e + 57586c6 commit 2e29414

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

mlxtend/feature_selection/tests/test_exhaustive_feature_selector.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,13 +9,14 @@
99
from numpy.testing import assert_almost_equal
1010
from packaging.version import Version
1111
from sklearn import __version__ as sklearn_version
12-
from sklearn.datasets import load_boston, load_iris
12+
from sklearn.datasets import load_iris
1313
from sklearn.ensemble import RandomForestClassifier
1414
from sklearn.linear_model import LinearRegression
1515
from sklearn.model_selection import GroupKFold
1616
from sklearn.neighbors import KNeighborsClassifier
1717

1818
from mlxtend.classifier import SoftmaxRegression
19+
from mlxtend.data import boston_housing_data
1920
from mlxtend.feature_selection import ExhaustiveFeatureSelector as EFS
2021
from mlxtend.utils import assert_raises
2122

@@ -334,8 +335,8 @@ def test_fit_params():
334335

335336

336337
def test_regression():
337-
boston = load_boston()
338-
X, y = boston.data[:, [1, 2, 6, 8, 12]], boston.target
338+
X, y = boston_housing_data()
339+
X = X[:, [1, 2, 6, 8, 12]]
339340
lr = LinearRegression()
340341
efs_r = EFS(
341342
lr,

0 commit comments

Comments
 (0)