Лучшая конфигурация после поиска гиперпараметров с помощью Ray Tune

Я только что запустил свой первый Ray Tune. Я получил хороший вывод на терминал и все это, но теперь мне интересно: какая конфигурация дала мне лучший результат?

Я вижу, что существует множество файлов результатов, но есть ли простой способ получить лучшую конфигурацию?


person xotix    schedule 09.05.2021    source источник


Ответы (1)


Вы можете использовать объект ExperimentAnalysis, возвращаемый tune.run(), чтобы получить наилучшую конфигурацию, например:

analysis = tune.run(trainable, search_alg=algo, stop={"training_iteration": 20})

best_trial = analysis.best_trial  # Get best trial
best_config = analysis.best_config  # Get best trial's hyperparameters
best_logdir = analysis.best_logdir  # Get best trial's logdir
best_checkpoint = analysis.best_checkpoint  # Get best trial's best checkpoint
best_result = analysis.best_result  # Get best trial's last results
best_result_df = analysis.best_result_df  # Get best result as pandas dataframe

См. Документацию: https://docs.ray.io/en/latest/tune/key-concepts.html#analysis.

person Kai    schedule 10.05.2021
comment
Ах да, я это видел. Запутался, но похоже, что этот объект анализа используется всеми запусками. - person xotix; 10.05.2021