如何从 h2o 中的 stackedsamble 中获取所有基本模型

Question

我在我的数据集上使用了 H2O 的 automl，最上面的模型是一个 stackedensemble 模型。我喜欢用这个堆叠模型的参数获取所有基本模型。我如何获得这些模型？

Answer 1

您可以使用 aml.leader 来获得顶级模型，或者您可以使用 get_best_model 来根据一些标准选择模型。如果你想从排行榜中选择一些特定的模型，你可以使用 h2o.get_model("model_id")（model_id 来自排行榜）。

要从所有基本模型中获取参数，您可以使用以下命令：

{base_model: h2o.get_model(base_model).actual_params
 for base_model in h2o.get_model("model_id").base_models}

您可能还想从堆叠集成和元学习器中获取参数：

metalearner_params = h2o.get_model("SE_model_id").metalearner().actual_params
se_params = h2o.get_model("SE_model_id").actual_params

How to get all base model from a stackedsamble in h2o