在H2O堆叠模型中获取特定模型

Question

所以我在H2O中使用GridSearch训练了几个GBM模型。

然后我训练了一个像这样的集合模型：

from h2o.estimators.stackedensemble import H2OStackedEnsembleEstimator

#List of all models in the GridSearch
all_model_ids=grid_search_gbm.model_ids


ensemble = H2OStackedEnsembleEstimator(
    model_id = 'ensemble_gbm' ,
    base_models = all_model_ids
)

ensemble.train(
    x=training_columns, 
    y=response_column,
    training_frame=train,
)

我无法从＆＃34; grid_search_gbm＆＃34;中提取特定模型。并将其输入＆＃34; H2OStackedEnsembleEstimator（）＆＃34;。有提取这个的方法吗？例如，＆＃39; grid_search_gbm＆＃39;？

中的3个最佳模型

Answer 1

你试过吗

grid_search_gbm.model_ids[:3]

它应该为您提供与

相同的模型ID

grid_search_gbm.sorted_metric_table()['model_ids'][:3]