python里随机森林的代码
from sklearn.ensemble import RandomForestRegressor
model = RandomForestRegressor()
model.fit(data.drop('y', axis=1), data[['y']])
但是出现了warning
A column-vector y was passed when a 1d array was expected
这个会影响预测结果吗?这个该怎么解决呢?
1个回答
这个warning应该不影响结果。你改成下面这样应该就没问题了
model.fit(data.drop('y', axis=1), data['y'])
或者
model.fit(data.drop('y', axis=1), data[['y']].values.flatten())
SofaSofa数据科学社区DS面试题库 DS面经