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

Question

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

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

Answer 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.