改成 SimpleImputer之後還是error?

2020/07/23 下午 11:17

機器學習共學討論版

Kevin

觀看數：6

回答數：2

收藏數：1

ml100

ml100-d21

問題:因為sklearn好像有更新說要換成from sklearn.impute import SimpleImputer,但我改了之後還是error如下

請問老師一下要如何修改

code:

from sklearn.preprocessing import MinMaxScaler

from sklearn.impute import SimpleImputer

# 特徵欄位清單

train = app_train

features = list(train.columns)

# 複製 test 資料

test = app_test.copy()

# 填補器 : 設定缺失值補中位數

imputer = SimpleImputer(strategy = 'median')

# 縮放器 : 設定特徵縮放到 0~1 區間

scaler = MinMaxScaler(feature_range = (0, 1))

# 填補器載入個欄中位數

# 將中位數回填 train, test 資料中的空缺值

train = imputer.transform(train)

test = imputer.transform(app_test)

# 縮放器載入 train 的上下限, 對 train, test 進行縮放轉換

train = scaler.transform(train)

test = scaler.transform(test)

print('Training data shape: ', train.shape)

print('Testing data shape: ', test.shape)