早期停车获取用于GBP的树木数量

Question

我使用早期停止和设置ntrees=10000训练了h2o中的GBM。我想检索模型中实际存在的树数。但是如果我打电话给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']（如果您想要列表中的数字），则另一种方法不需要计算最大但仍然很难做到model.summary()。或者只是 Table "public.goods" Column | Type | Modifiers -----------------+--------------------------+--------------------------- goods_id | character varying | not null goods_owner_id | character varying | not null ，如果您只想查看数字。