我正在尝试在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