问题描述
我刚刚运行了我的第一个 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}}