From 2af8404a761ab377d51224dbd1b6b921348d3d24 Mon Sep 17 00:00:00 2001 From: Hanmin Qin Date: Wed, 11 Jul 2018 21:28:48 +0800 Subject: [PATCH] TST Use SimpleImputer instead of Imputer in test_bagging.py --- sklearn/ensemble/tests/test_bagging.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/sklearn/ensemble/tests/test_bagging.py b/sklearn/ensemble/tests/test_bagging.py index 626b34f58e5a6..ec815e8ec93af 100644 --- a/sklearn/ensemble/tests/test_bagging.py +++ b/sklearn/ensemble/tests/test_bagging.py @@ -33,7 +33,7 @@ from sklearn.model_selection import train_test_split from sklearn.datasets import load_boston, load_iris, make_hastie_10_2 from sklearn.utils import check_random_state -from sklearn.preprocessing import Imputer +from sklearn.impute import SimpleImputer from scipy.sparse import csc_matrix, csr_matrix @@ -777,9 +777,9 @@ def test_bagging_regressor_with_missing_inputs(): for y in y_values: regressor = DecisionTreeRegressor() pipeline = make_pipeline( - Imputer(), - Imputer(missing_values=np.inf), - Imputer(missing_values=np.NINF), + SimpleImputer(), + SimpleImputer(missing_values=np.inf), + SimpleImputer(missing_values=np.NINF), regressor ) pipeline.fit(X, y).predict(X) @@ -807,9 +807,9 @@ def test_bagging_classifier_with_missing_inputs(): y = np.array([3, 6, 6, 6, 6]) classifier = DecisionTreeClassifier() pipeline = make_pipeline( - Imputer(), - Imputer(missing_values=np.inf), - Imputer(missing_values=np.NINF), + SimpleImputer(), + SimpleImputer(missing_values=np.inf), + SimpleImputer(missing_values=np.NINF), classifier ) pipeline.fit(X, y).predict(X)