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

  统计/机器学习 监督式学习 Python    浏览次数:211        分享
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



  相关主题

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

决策树是如何得到量化的概率,不只是0或1的标签?   2回答

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

决策树算法ID3,C4.5和CART的特点、异同?   3回答

python sklean中的决策树是用的哪一种决策树算法?   2回答

随机森林中增加树的数量对于偏差和方差的影响   2回答

Extra Tree算法   1回答

剪枝是什么意思   1回答

关于knn算法中kd树的问题   1回答

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

决策树、随机森林中的多重共线性问题   3回答

决策树的深度和数据特征个数的关系   1回答



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

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

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

我们谢绝答非所问。

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

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