add config

2024-09-14 20:30:44 +02:00
parent f9b17473cb
commit a00f9caaf7
5 changed files with 226 additions and 0 deletions
@@ -0,0 +1,40 @@
+program: ./scripts/train.py
+project: wne-masters-thesis-testing
+command:
+  - ${env}
+  - ${interpreter}
+  - ${program}
+  - "./configs/experiments/informer-btcusdt-5m-quantile.yaml"
+  - "--patience"
+  - "10"
+method: random
+metric:
+  goal: minimize
+  name: val_loss
+parameters:
+  past_window:
+    distribution: int_uniform
+    min: 20
+    max: 120
+  batch_size:
+    values: [64, 128, 256]
+  model:
+    parameters:
+      name:
+        value: "Informer"
+      dmodel:
+        values: [256, 512, 1024]
+      d_fully_connected:
+        values: [256, 512, 1024]
+      n_attention_heads:
+        values: [1, 2, 4, 6]
+      dropout:
+        values: [0.05, 0.1, 0.2, 0.3]
+      n_encoder_layers:
+        values: [1, 2, 4, 6]
+      n_decoder_layers: 
+        values: [1, 2]
+      learning_rate:
+        values: [0.001, 0.0005, 0.0001]
+      optimizer:
+        value: "Adam"