codes.tune.evaluate_study

Functions

load_model_test_losses(model_path)	Load the test losses from the model checkpoint.
load_study_config(study_name)	Load the YAML config used by the study (optuna_config.yaml).
main()	Main function to evaluate an Optuna study and its top models.
moving_average(data, window_size)	Compute the moving average of a 1D array.
parse_arguments()	Parse command-line arguments.
plot_test_losses(test_losses, labels, study_name)	Plot the test losses of the top models.