review

charlottecvn · Feb 13, 2024 · e3edd4f · e3edd4f
1 parent 49f2a9a
commit e3edd4f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/eval/bayesian_opt.py b/eval/bayesian_opt.py
@@ -240,6 +240,10 @@ def objective(trial, n_trial, merged_dataset, data_order, txt_name):
     study = optuna.create_study(direction='maximize', storage="sqlite:///db.sqlite3", study_name=txt_optuna)
     study.optimize(lambda trial: objective(trial, trials, merged_dataset, data_order, txt_name), n_trials=trials)
 
+    optuna.plot_intermediate_values(study)
+    optuna.plot_parallel_coordinate(study)
+    optuna.plot_contour(study)
+
     print('Number of finished trials:', len(study.trials))
     print('Best trial:', study.best_trial.params)
     print(study.best_trial.number)