如何知道在XGBoost中创建的树的数量

How to know the number of tree created in XGBoost(如何知道在XGBoost中创建的树的数量)

问题描述

我有一个关于XGBoost的问题。

您知道如何知道在XGBoost中创建的树的数量吗？ 与RandomForest不同，模型制造商决定制作多少棵树，XGBoost基本上继续创建树，直到损失函数达到一定的数字。因此我想知道这一点。

谢谢。

推荐答案

它有点歪曲，但我现在做的是dump-模型(XGBoost生成一个列表，其中每个元素都是单个树的字符串表示)，然后计算列表中有多少个元素：

# clf is a XGBoost model fitted using the sklearn API
dump_list = clf.get_booster().get_dump()
num_trees = len(dump_list)

这篇关于如何知道在XGBoost中创建的树的数量的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！