Python中的H2O GridSearch导出结果

时间:2019-06-04 21:13:29

标签: python h2o grid-search

我正在尝试在Python API中使用H2O的GridSearch进行超参数调整。我有兴趣导出所有尝试的组合的完整报告。与'grid_search.get_grid()。models'命令打印的报告非常相似。但是,我无法找到一种方法将此报告导出到数据框中。

这是我正在使用的代码:

estimator = H2ORandomForestEstimator( balance_classes = True)

hyper_parameters = {'ntrees':[10 ,50, 100], 'max_depth':[2, 5, 10, 20]}

grid_search = H2OGridSearch(model = estimator, hyper_params = hyper_parameters)

grid_search.train(x = feature_columns, y = response_column, training_frame = train_h2o, nfolds = 5)

grid_search.get_grid().models

0 个答案:

没有答案