python - 获取用于早期停止的 GBM 的树的数量

Question

我使用提前停止和设置 ntrees=10000 在水中训练了 GBM。我想检索模型中实际存在的树木数量。但是，如果我调用 model.params['ntrees'] (其中 model 是网格搜索中的最佳模型)，我会得到

{'default': 50, 'actual': 10000}

其中 10000 是我在训练期间设置的参数，但不是模型中最终出现的树的实际数量。

如果我调用 model.score_history()，那么我可以看到提前停止在 280 树处开始。但肯定有比这个 hack 更直接的方法来找出模型中树木的实际数量:

best_model.score_history()['number_of_trees'].max()

Answer 1

目前还没有一种干净的方法可以做到这一点。另一种方法是 model.summary()['number_of_trees'][0]，不需要计算最大值，但仍然很笨拙，如果您想要数字，model. Summary()['number_of_trees'] 如果您想要列表中的数字。或者如果您只想查看数字，则只需 model.summary() 即可。