获得提前停止的GBM使用的树木数量。

Question

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

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

其中 10000 是我在训练过程中设置的参数，但不是最终进入模型的实际树木数量。

如果我调用 model.score_history() 那么，我可以看到，早期的停止踢在 280 树木的数量。但肯定有比这个黑客更直接的方法来找出模型中的实际树数。

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

Answer 1

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