使用 Ray Tune 进行超参数搜索后的最佳配置

问题描述

我刚刚运行了我的第一个 Ray Tune。我得到了不错的终端输出,但现在我想知道:哪种配置给了我最好的分数?

我看到有大量的结果文件,但有没有一种简单的方法来获得最佳配置?

解决方法

您可以使用 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

查看文档:{​​{3}}