H2O GridSearch в Python, экспортирующий результаты - PullRequest
0 голосов
/ 05 июня 2019

Я пытаюсь использовать H2O GridSearch в Python API для настройки гиперпараметров.Я заинтересован экспортировать полный отчет по всем опробованным комбинациям.Нечто похожее на отчет, который печатается командой 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
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...