本文介绍了如何知道在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中创建的树的数量的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！