experiment directory change

ITMO-NSS-team · Oct 25, 2023 · add3659 · add3659
1 parent 7d93282
commit add3659
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -134,6 +134,3 @@ data/
 .idea/
 .run/
 .vscode/
-
-# Experiment folder
-/experiments
diff --git a/configs/train_surrogate_TS.yml b/configs/train_surrogate_TS.yml
@@ -39,7 +39,7 @@ trainer:
   devices: "auto"
 
 tensorboard_logger:
-  save_dir: "./experiments/TS"
+  save_dir: "./logs/TS"
   name: "model_training"
 
 model_checkpoint_callback:

diff --git a/configs/train_surrogate_model.yml b/configs/train_surrogate_model.yml
@@ -22,7 +22,7 @@ model:
     global_pool: "mean"
     use_edge_attr: False
     pipe_encoder_type: "graph_transformer"
-    dataset_encoder_type: "aggregated"  #"column"
+    dataset_encoder_type: "column"  #"column" or "aggregated"
   weight_decay: 0.0001
   lr: 0.001
   temperature: 5
@@ -42,7 +42,7 @@ trainer:
   devices: "auto"
 
 tensorboard_logger:
-  save_dir: "./experiments/openml_meta_features_and_fedot_pipelines/train_surrogate_model_best"
+  save_dir: "./logs/openml_meta_features_and_fedot_pipelines/train_surrogate_model_best"
   name: "model_training"
 
 model_checkpoint_callback:

diff --git a/configs/tune_surrogate_model.yml b/configs/tune_surrogate_model.yml
@@ -43,7 +43,7 @@ trainer:
   devices: "auto"
 
 tensorboard_logger:
-  save_dir: "./experiments/tune_surrogate_model"
+  save_dir: "./logs/tune_surrogate_model"
   name: "hyperparameter_tuning"
 
 model_checkpoint_callback: