Skip to content

Commit

Permalink
review
Browse files Browse the repository at this point in the history
  • Loading branch information
charlottecvn committed Feb 16, 2024
1 parent 95248b6 commit 5a55989
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions eval/hyperopt_gat.py
Original file line number Diff line number Diff line change
Expand Up @@ -267,8 +267,8 @@ def objective(trial, n_trial, merged_dataset, data_order, txt_name):
print(f"Trial finished with test loss {loss} and test accuracy {accuracy}")
return accuracy

#storage = JournalStorage(JournalFileStorage("/ceph/knmimo/GNNs_UQ_charlotte/GINenergygrids/slurm/optuna-gat.log"))
storage = f"sqlite:///db.gat.sqlite3"#"sqlite:///gat-study.db"
storage = JournalStorage(JournalFileStorage("/ceph/knmimo/GNNs_UQ_charlotte/GINenergygrids/slurm/optuna-gat.log"))
#storage = f"sqlite:///db.gat.sqlite3"#"sqlite:///gat-study.db"

study = optuna.create_study(
direction="maximize", storage=storage, study_name=txt_optuna
Expand Down

0 comments on commit 5a55989

Please sign in to comment.