aepsa_v2 solarenergy

.vscode/launch.json

@@ -225,6 +225,14 @@
             "program": "run.py",
             "console": "integratedTerminal",
             "args": "--config ./config/AEPSA/v2_METR-LA.yaml"
-        }
+        },
+        {
+            "name": "AEPSA_v2: SolarEnergy",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "run.py",
+            "console": "integratedTerminal",
+            "args": "--config ./config/AEPSA/v2_SolarEnergy.yaml"
+        },
     ]
 }

config/AEPSA/v2_SolarEnergy.yaml

@@ -0,0 +1,59 @@
+basic:
+  dataset: "SolarEnergy"
+  mode : "train"
+  device : "cuda:1"
+  model: "AEPSA_v2"
+  seed: 2023
+
+data:
+  add_day_in_week: true
+  add_time_in_day: true
+  column_wise: false
+  days_per_week: 7
+  default_graph: true
+  horizon: 24
+  lag: 24
+  normalizer: std
+  num_nodes: 137
+  steps_per_day: 24
+  test_ratio: 0.2
+  tod: false
+  val_ratio: 0.2
+  sample: 1
+  input_dim: 1
+  batch_size: 64
+
+model:
+  pred_len: 24
+  seq_len: 24
+  patch_len: 6 
+  stride: 7
+  dropout: 0.2
+  gpt_layers: 9
+  d_ff: 128
+  gpt_path: ./GPT-2
+  d_model: 64
+  n_heads: 1
+  input_dim: 1
+  word_num: 1000
+  num_nodes: 137
+
+train:
+  batch_size: 64
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: "5,20,40,70"
+  lr_init: 0.003
+  max_grad_norm: 5
+  weight_decay: 0
+  debug: false
+  output_dim: 1
+  log_step: 100
+  plot: false
+  mae_thresh: None
+  mape_thresh: 0.001