如何知道在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)