File tree Expand file tree Collapse file tree 1 file changed +3
-6
lines changed
mlxtend/feature_selection/tests Expand file tree Collapse file tree 1 file changed +3
-6
lines changed Original file line number Diff line number Diff line change 33# Author: Sebastian Raschka <sebastianraschka.com>
44#
55# License: BSD 3 clause
6- from mlxtend .data import boston_housing_data
76import numpy as np
8- from numpy import nan
97from numpy .testing import assert_almost_equal
108from sklearn .datasets import load_iris
119from sklearn .ensemble import RandomForestClassifier
1210from sklearn .linear_model import LinearRegression
1311from sklearn .metrics import roc_auc_score
1412from sklearn .neighbors import KNeighborsClassifier
1513
14+ from mlxtend .data import boston_housing_data
1615from mlxtend .feature_selection import SequentialFeatureSelector as SFS
1716from mlxtend .utils import assert_raises
1817
@@ -163,8 +162,7 @@ def test_keyboard_interrupt():
163162
164163
165164def test_max_feature_subset_best ():
166- boston = load_boston ()
167- X , y = boston .data , boston .target
165+ X , y = boston_housing_data
168166 lr = LinearRegression ()
169167
170168 sfs = SFS (
@@ -189,8 +187,7 @@ def test_max_feature_subset_best():
189187
190188
191189def test_max_feature_subset_parsimonious ():
192- boston = load_boston ()
193- X , y = boston .data , boston .target
190+ X , y = boston_housing_data
194191 lr = LinearRegression ()
195192
196193 sfs = SFS (
You can’t perform that action at this time.
0 commit comments