h2o - 如何获取由 h2o automl 生成的非领导者模型的详细信息？

Question

运行后automl（3类分类），可以看到模型列表如下： model_id mean_per_class_error StackedEnsemble_BestOfFamily_0_AutoML_20180420_174925 0.262355 StackedEnsemble_AllModels_0_AutoML_20180420_174925 0.262355 XRT_0_AutoML_20180420_174925 0.266606 DRF_0_AutoML_20180420_174925 0.278428 GLM_grid_0_AutoML_20180420_174925_model_0 0.442917

但mean_per_class_error对于我的情况来说，这不是一个好的指标，因为班级不平衡（一个班级的人口非常少）。如何获取非领导者模型的详细信息并计算其他指标？谢谢。

蟒蛇版本：3.6.0

h2o 版本：3.18.0.5

Answer 1

实际上只是我自己想通了（假设aml是训练后的 h2o automl 对象）：

for m in aml.leaderboard.as_data_frame()['model_id']: print(m) print(h2o.get_model(m))

Answer 2

您还可以使用以下行获取您感兴趣的相应模型：

model6 = h2o.get_model(aml.leaderboard.as_data_frame()['model_id'][6])

其中 6 是模型在排行榜中的索引号。