如何获取RandomForestRegressor中的所有集合估计(scikit-learn)

时间:2013-04-20 16:00:16

标签: machine-learning scikit-learn random-forest ensemble-learning

我正在尝试进行随机森林回归,我希望通过查看整体中每个回归树的输出来获得我的估计分布,并以某种列表返回给我。在R中,在predict.all预测方法中提供randomForest选项可以为我做这个,有没有办法在sklearn的RandomForestRegressor类中执行此操作?

在0.13及以上版本中,我看到返回叶子索引的方法apply,但是我不知道如何使用这些,我在RandomForestRegressor的文档中没有看到任何更多的线索。

谢谢你的帮助。

1 个答案:

答案 0 :(得分:2)

你能升级到0.14吗?在0.14中,有一个estimators_成员,可以访问各个树。请参阅RandomForestRegressor文档。