ValueError: Expected 2D array, got 1D array instead

从 Warning 信息中得知,原因是 sklearn 的新版本中,OneHotEncoder 的输入必须是 2-D array,而 data_train['Fare'] 返回的 Series 本质上是 1-D array,所以要将

df['Fare_scaled'] = scaler.fit_transform(data_train['Fare'])

改成

df['Fare_scaled'] = scaler.fit_transform(data_train[['Fare']])

也就是在data_train[['Fare']]外面再加一个[]

转自:http://blog.sina.com.cn/s/blog_4fa881040102y6pz.html

你可能感兴趣的:(ValueError: Expected 2D array, got 1D array instead)