修复配置, trainer

.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/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

config/TWDGCN/PEMSD4.yaml

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