运行后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
实际上只是自己想通了(假设aml是训练后的 h2o automl 对象):
for m in aml.leaderboard.as_data_frame()['model_id']:
print(m)
print(h2o.get_model(m))