STID配置更新

更新main支的修复

.vscode/launch.json

@@ -4,14 +4,21 @@
     // 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387
     "version": "0.2.0",
     "configurations": [
-
 {
-            "name": "EXP_PEMSD8",
+            "name": "STID_PEMS-BAY",
             "type": "debugpy",
             "request": "launch",
             "program": "run.py",
             "console": "integratedTerminal",
-            "args": "--config ./config/DDGCRN/PEMSD8.yaml"
+            "args": "--config ./config/STID/PEMS-BAY.yaml"
+        },
+        {
+            "name": "STID_PEMSD4",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "run.py",
+            "console": "integratedTerminal",
+            "args": "--config ./config/STID/PEMSD4.yaml"
         },
         {
             "name": "REPST",

.vscode/settings.json

@@ -0,0 +1,5 @@
+{
+    "python-envs.defaultEnvManager": "ms-python.python:conda",
+    "python-envs.defaultPackageManager": "ms-python.python:conda",
+    "python-envs.pythonProjects": []
+}

config/AGCRN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "AGCRN"
+
 data:
   num_nodes: 358
   lag: 12

config/AGCRN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "AGCRN"
+
 data:
   num_nodes: 307
   lag: 12

config/AGCRN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "AGCRN"
+
 data:
   num_nodes: 883
   lag: 12

config/AGCRN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "AGCRN"
+
 data:
   num_nodes: 170
   lag: 12

config/ARIMA/Hainan.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "Hainan"
+  mode: "train"
+  device: "cuda:0"
+  model: "ARIMA"
+
 data:
   num_nodes: 13
   lag: 12

config/ARIMA/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "ARIMA"
+
 data:
   num_nodes: 358
   lag: 12

config/ARIMA/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "ARIMA"
+
 data:
   num_nodes: 307
   lag: 12

config/ARIMA/PEMSD7(L).yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7(L)"
+  mode: "train"
+  device: "cuda:0"
+  model: "ARIMA"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/ARIMA/PEMSD7(M).yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7(M)"
+  mode: "train"
+  device: "cuda:0"
+  model: "ARIMA"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/ARIMA/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "ARIMA"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/ARIMA/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "ARIMA"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/DCRNN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "DCRNN"
+
 data:
   num_nodes: 358
   lag: 12

config/DCRNN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "DCRNN"
+
 data:
   num_nodes: 307
   lag: 12

config/DCRNN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "DCRNN"
+
 data:
   num_nodes: 883
   lag: 12

config/DCRNN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "DCRNN"
+
 data:
   num_nodes: 170
   lag: 12

config/DDGCRN/Hainan.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "Hainan"
+  mode: "train"
+  device: "cuda:0"
+  model: "DDGCRN"
+
 data:
   num_nodes: 13
   lag: 12

config/DDGCRN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "DDGCRN"
+
 data:
   num_nodes: 358
   lag: 12

config/DDGCRN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "DDGCRN"
+
 data:
   num_nodes: 307
   lag: 12

config/DDGCRN/PEMSD7(L).yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7(L)"
+  mode: "train"
+  device: "cuda:0"
+  model: "DDGCRN"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/DDGCRN/PEMSD7(M).yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7(M)"
+  mode: "train"
+  device: "cuda:0"
+  model: "DDGCRN"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/DDGCRN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "DDGCRN"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/DSANET/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "DSANET"
+
 data:
   num_nodes: 358
   lag: 12

config/DSANET/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "DSANET"
+
 data:
   num_nodes: 307
   lag: 12

config/DSANET/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "DSANET"
+
 data:
   num_nodes: 883
   lag: 12

config/DSANET/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "DSANET"
+
 data:
   num_nodes: 170
   lag: 12

config/EXP/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "EXP"
+
 
 
 data:

config/EXP/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "EXP"
+
 data:
   num_nodes: 307
   lag: 12

config/EXP/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "EXP"
+
 data:
   num_nodes: 883
   lag: 12

config/EXP/SD.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "SD"
+  mode: "train"
+  device: "cuda:0"
+  model: "EXP"
+
 data:
   num_nodes: 716
   lag: 12

config/EXPB/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "EXPB"
+
 data:
   num_nodes: 307
   lag: 12

config/GWN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "GWN"
+
 data:
   num_nodes: 358
   lag: 12

config/GWN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "GWN"
+
 data:
   num_nodes: 307
   lag: 12

config/GWN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "GWN"
+
 data:
   num_nodes: 883
   lag: 12

config/GWN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "GWN"
+
 data:
   num_nodes: 170
   lag: 12

config/MegaCRN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "MegaCRN"
+
 data:
   num_nodes: 358
   lag: 12

config/MegaCRN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "MegaCRN"
+
 data:
   num_nodes: 307
   lag: 12

config/MegaCRN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "MegaCRN"
+
 data:
   num_nodes: 883
   lag: 12

config/MegaCRN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "MegaCRN"
+
 data:
   num_nodes: 170
   lag: 12

config/NLT/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "NLT"
+
 data:
   num_nodes: 358
   lag: 12

config/NLT/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "NLT"
+
 data:
   num_nodes: 307
   lag: 12

config/NLT/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "NLT"
+
 data:
   num_nodes: 883
   lag: 12

config/NLT/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "NLT"
+
 data:
   num_nodes: 170
   lag: 12

config/PDG2SEQ/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "PDG2SEQ"
+
 data:
   num_nodes: 358
   lag: 12

config/PDG2SEQ/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "PDG2SEQ"
+
 data:
   num_nodes: 307
   lag: 12

config/PDG2SEQ/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "PDG2SEQ"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/PDG2SEQ/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "PDG2SEQ"
+
 data:
   add_day_in_week: true
   add_time_in_day: true

config/STAEFormer/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STAEFormer"
+
 data:
   num_nodes: 358
   lag: 12

config/STAEFormer/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STAEFormer"
+
 data:
   num_nodes: 307
   lag: 12

config/STAEFormer/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STAEFormer"
+
 data:
   num_nodes: 883
   lag: 12

config/STAEFormer/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STAEFormer"
+
 data:
   num_nodes: 170
   lag: 12

config/STAWnet/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STAWnet"
+
 data:
   num_nodes: 358
   lag: 12

config/STAWnet/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STAWnet"
+
 data:
   num_nodes: 307
   lag: 12

config/STAWnet/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STAWnet"
+
 data:
   num_nodes: 883
   lag: 12

config/STAWnet/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STAWnet"
+
 data:
   num_nodes: 170
   lag: 12

config/STFGNN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STFGNN"
+
 data:
   num_nodes: 358
   lag: 12

config/STFGNN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STFGNN"
+
 data:
   num_nodes: 307
   lag: 12

config/STFGNN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STFGNN"
+
 data:
   num_nodes: 883
   lag: 12

config/STFGNN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STFGNN"
+
 data:
   num_nodes: 170
   lag: 12

config/STGCN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGCN"
+
 data:
   num_nodes: 358
   lag: 12

config/STGCN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGCN"
+
 data:
   num_nodes: 307
   lag: 12

config/STGCN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGCN"
+
 data:
   num_nodes: 883
   lag: 12

config/STGCN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGCN"
+
 data:
   num_nodes: 170
   lag: 12

config/STGNCDE/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGNCDE"
+
 data:
   num_nodes: 358
   lag: 12

config/STGNCDE/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGNCDE"
+
 data:
   num_nodes: 307
   lag: 12

config/STGNCDE/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGNCDE"
+
 data:
   num_nodes: 883
   lag: 12

config/STGNCDE/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGNCDE"
+
 data:
   num_nodes: 170
   lag: 12

config/STGNRDE/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGNRDE"
+
 data:
   num_nodes: 358
   lag: 12

config/STGNRDE/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGNRDE"
+
 data:
   num_nodes: 307
   lag: 12

config/STGNRDE/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGNRDE"
+
 data:
   num_nodes: 883
   lag: 12

config/STGNRDE/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGNRDE"
+
 data:
   num_nodes: 170
   lag: 12

config/STGODE/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGODE"
+
 data:
   num_nodes: 358
   lag: 12

config/STGODE/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGODE"
+
 data:
   num_nodes: 307
   lag: 12

config/STGODE/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGODE"
+
 data:
   num_nodes: 883
   lag: 12

config/STGODE/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STGODE"
+
 data:
   num_nodes: 170
   lag: 12

config/STID/PEMS-BAY.yaml

@@ -0,0 +1,66 @@
+basic:
+  dataset: "PEMS-BAY"
+  mode: "train"
+  device: "cuda:0"
+  model: "STID"
+
+data:
+  num_nodes: 325
+  lag: 24
+  horizon: 24
+  val_ratio: 0.2
+  test_ratio: 0.2
+  tod: False
+  normalizer: std
+  column_wise: False
+  default_graph: True
+  add_time_in_day: True
+  add_day_in_week: True
+  steps_per_day: 288
+  days_per_week: 7
+  input_dim: 1
+  output_dim: 1
+  batch_size: 64
+
+model:
+  input_dim: 3
+  output_dim: 1
+  history: 24
+  horizon: 24
+  num_nodes: 325
+  input_len: 24
+  embed_dim: 32
+  output_len: 24
+  num_layer: 3
+  if_node: True
+  node_dim: 32
+  if_T_i_D: True
+  if_D_i_W: True
+  temp_dim_tid: 32
+  temp_dim_diw: 32
+  time_of_day_size: 288
+  day_of_week_size: 7
+  batch_size: 64
+
+
+train:
+  loss_func: mae
+  seed: 1
+  batch_size: 64
+  epochs: 300
+  lr_init: 0.002
+  weight_decay: 0.0001
+  lr_decay: False
+  lr_decay_rate: 0.3
+  lr_decay_step: "1,50,80"
+  early_stop: True
+  early_stop_patience: 15
+  grad_norm: False
+  max_grad_norm: 5
+  real_value: True
+  debug: true
+  output_dim: 1
+  mae_thresh: null
+  mape_thresh: 0.0
+  log_step: 200
+  plot: False

config/STID/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STID"
+
 data:
   num_nodes: 307
   lag: 12

config/STIDGCN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STIDGCN"
+
 data:
   num_nodes: 358
   lag: 12

config/STIDGCN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STIDGCN"
+
 data:
   num_nodes: 307
   lag: 12

config/STIDGCN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STIDGCN"
+
 data:
   num_nodes: 883
   lag: 12

config/STIDGCN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STIDGCN"
+
 data:
   num_nodes: 170
   lag: 12

config/STMLP/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STMLP"
+
 data:
   num_nodes: 358
   lag: 12

config/STMLP/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STMLP"
+
 data:
   num_nodes: 307
   lag: 12

config/STMLP/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STMLP"
+
 data:
   num_nodes: 883
   lag: 12

config/STMLP/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STMLP"
+
 data:
   num_nodes: 170
   lag: 12

config/STSGCN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "STSGCN"
+
 data:
   num_nodes: 358
   lag: 12

config/STSGCN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "STSGCN"
+
 data:
   num_nodes: 307
   lag: 12

config/STSGCN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "STSGCN"
+
 data:
   num_nodes: 883
   lag: 12

config/STSGCN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "STSGCN"
+
 data:
   num_nodes: 170
   lag: 12

config/ST_SSL/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "ST_SSL"
+
 data:
   num_nodes: 358
   lag: 12

config/ST_SSL/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "ST_SSL"
+
 data:
   num_nodes: 307
   lag: 12

config/ST_SSL/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "ST_SSL"
+
 data:
   num_nodes: 883
   lag: 12

config/ST_SSL/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "ST_SSL"
+
 data:
   num_nodes: 170
   lag: 12

config/TCN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "TCN"
+
 data:
   num_nodes: 358
   lag: 12

config/TCN/PEMSD4.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD4"
+  mode: "train"
+  device: "cuda:0"
+  model: "TCN"
+
 data:
   num_nodes: 307
   lag: 12

config/TCN/PEMSD7.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD7"
+  mode: "train"
+  device: "cuda:0"
+  model: "TCN"
+
 data:
   num_nodes: 883
   lag: 12

config/TCN/PEMSD8.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD8"
+  mode: "train"
+  device: "cuda:0"
+  model: "TCN"
+
 data:
   num_nodes: 170
   lag: 12

config/TWDGCN/Hainan.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "Hainan"
+  mode: "train"
+  device: "cuda:0"
+  model: "TWDGCN"
+
 data:
   num_nodes: 13
   lag: 12

config/TWDGCN/PEMSD3.yaml

@@ -1,3 +1,9 @@
+basic:
+  dataset: "PEMSD3"
+  mode: "train"
+  device: "cuda:0"
+  model: "TWDGCN"
+
 data:
   num_nodes: 358
   lag: 12