我正在尝试拟合一个随机森林回归模型,并且希望通过查看集成中每个回归树的输出,获得我的估计分布,并以某种列表形式返回。在R语言中,通过在randomForest
的预测方法中提供predict.all
选项可以实现这一点。在sklearn的RandomForestRegressor
类中,有没有类似的方法可以做到这一点?
在0.13及以上版本中,我看到了apply
方法,它返回叶子节点的索引,但我不知道如何使用这些索引,并且在RandomForestRegressor
的文档中也没有找到更多线索。
感谢您的帮助。
回答:
您可以升级到0.14版本吗?在0.14版本中,有一个estimators_
成员,可以让您访问各个树。请查看RandomForestRegressor的文档。