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

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

这个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



  相关主题

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

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

Extra Tree算法   1回答

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

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

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

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

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

如果采用非等权重的方法集成一个随机森林,会表现得更好嘛?   1回答

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

剪枝是什么意思   1回答

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



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

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

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

我们谢绝答非所问。

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

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