使用XGBoost随机森林，如何从每棵树中提取预测？

发布于 2025-02-07 16:30:08 字数 423 浏览 1 评论 0 原文

使用Sklearn Randomforest，您可以像这样的每棵树的每棵树的预测：

per_tree_pred = [tree.predict(X) for tree in clf.estimators_]

xgboost作为安装随机森林的选项，这很不错，因为我们可以利用GPU，但是有什么方法可以像我们在Sklearn中一样从每棵树上获得预测？这对于了解预测的不确定性可能非常有用。

谢谢，

原文

With the sklearn RandomForest, you can get the prediction of each tree of the random forest like so:

per_tree_pred = [tree.predict(X) for tree in clf.estimators_]

xgboost as the option of fitting random forest, which is quite nice since we can leverage the GPU, but is there any way to get the prediction from each tree like we can in sklearn? This can be very useful to get a sense of the uncertainty of the predictions.

Thanks,

分享到QQ

分享到微博