随机森林警告:A column-vector y was passed when a 1d array was expected

  统计/机器学习 监督式学习 Python    浏览次数:423        分享
0

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

这个会影响预测结果吗?这个该怎么解决呢?

 

董咚咚   2019-04-13 04:34



   1个回答 
3

这个warning应该不影响结果。你改成下面这样应该就没问题了

model.fit(data.drop('y', axis=1), data['y'])

或者

model.fit(data.drop('y', axis=1), data[['y']].values.flatten())
SofaSofa数据科学社区DS面试题库 DS面经

u_u   2019-04-13 08:07



  相关主题

怎么得到randomforestregressor里每棵树的预测结果?   1回答

随机森林的模型一般都很大么?   1回答

决策树剪枝有什么策略或者注意事项?   2回答

剪枝是什么意思   1回答

python里实现一个简单的决策树   1回答

Adaboost里的树有没有可能权重是负数?   1回答

随机森林回归   1回答

为什么随机森林比决策树更robust?   1回答

决策树可以做多元分类吗?   1回答

Gradient tree boosting和random forest (随机森林) 有什么区别和联系   1回答

怎么理解随机森林的oob error?   2回答

随机森林是如何计算测试样本的概率的?   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!