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

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



  相关讨论

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

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

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

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

lightgbm使用自定义损失函数后,二分类的预测结果不再是【0,1】区间

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

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

决策树怎么做增量学习或者online学习?

Extra Tree算法

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

  随便看看

序列的autocorrelation(自相关系数)的计算公式

xgboost怎么调参?

怎么理解tweedie分布?

为什么神经网络模型不用交叉验证?

如何重命名pandas的dataframe的列名