【发布时间】:2022-08-09 23:55:56
【问题描述】:
我无法理解 H20 automl 生成的模型!
输出是这样的,例如: StackedEnsemble_AllModels_1_AutoML_1_20220809_134944
我怎么知道堆叠使用了哪些基本模型?
我无法理解 H20 automl 生成的模型!
输出是这样的,例如: StackedEnsemble_AllModels_1_AutoML_1_20220809_134944
我怎么知道堆叠使用了哪些基本模型?
假设您在名为aml 的变量中有一个H2OAutoML 对象,然后您可以使用aml.leaderboard 查看排行榜。排行榜基本上是按某些指标(AUC、RMSE 等)排序的所有训练模型的表格。
堆叠集成使用在它们之前训练的模型 - 所有模型或每个模型系列(GBM、GLM、DRF 等)的最佳模型。要查看基本模型,您可以使用以下内容:
print(aml.leaderboard) # to see what models were trained
# Let's assume you like the SE model with
# model_id "StackedEnsemble_AllModels_1_AutoML_2_20220809_174830"
# You can retrieve the model by using h2o.get_model:
se = h2o.get_model("StackedEnsemble_AllModels_1_AutoML_2_20220809_174830")
# And then you can list model ids of the base models:
se.base_models
【讨论】: