codes.tune.evaluate_tuning

codes.tune.evaluate_tuning#

Functions

evaluate_tuning(study_prefix[, top_n])

For all surrogate studies named '<study_prefix>_<surrogate>', plot the top_n test-loss trajectories.

get_best_trials(study, top_n)

Return trial numbers of best 'top_n' trials.

load_loss_history(model_path)

Load loss histories from a saved model file (.pth).

main()

parse_args()

plot_losses(loss_histories, epochs, labels)

Plot multiple loss trajectories on a log scale.