.vscode/launch.json

@@ -6,233 +6,12 @@
     
     "configurations": [
         {
-            "name": "DDGCRN: METR-LA",
+            "name": "train",
             "type": "debugpy",
             "request": "launch",
-            "program": "run.py",
+            "program": "train.py",
             "console": "integratedTerminal",
-            "args": "--config ./config/DDGCRN/METR-LA.yaml"
+            "justMyCode": false
-        },
+        }
-        // STID 模型组
-        {
-            "name": "STID: PEMS-BAY",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/STID/PEMS-BAY.yaml"
-        },
-        {
-            "name": "STID: METR-LA",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/STID/METR-LA.yaml"
-        },
-        {
-            "name": "STID: PEMSD4",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/STID/PEMSD4.yaml"
-        },
-        {
-            "name": "STID: BJTaxi-InFlow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/STID/BJTaxi_Inflow.yaml"
-        },
-        {
-            "name": "STID: BJTaxi-OutFlow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/STID/BJTaxi_Outflow.yaml"
-        },
-        {
-            "name": "STID: NYCBike-InFlow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/STID/NYCBike_Inflow.yaml"
-        },
-        {
-            "name": "STID: NYCBike-OutFlow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/STID/NYCBike_Outflow.yaml"
-        },
-        {
-            "name": "STID: SolarEnergy",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/STID/SolarEnergy.yaml"
-        },
-        
-        // REPST 模型组
-        {
-            "name": "REPST: PEMSD8",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/PEMSD8.yaml"
-        },
-        {
-            "name": "REPST: BJTaxi-InFlow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/BJTaxi-Inflow.yaml"
-        },
-        {
-            "name": "REPST: NYCBike-outflow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/NYCBike-outflow.yaml"
-        },
-        {
-            "name": "REPST: NYCBike-inflow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/NYCBike-inflow.yaml"
-        },
-        {
-            "name": "REPST: PEMS-BAY",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/PEMS-BAY.yaml"
-        },
-        {
-            "name": "REPST: METR-LA",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/METR-LA.yaml"
-        },
-        {
-            "name": "REPST: SolarEnergy",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/SolarEnergy.yaml"
-        },
-        {
-            "name": "REPST: BeijingAirQuality",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/BeijingAirQuality.yaml"
-        },
-        {
-            "name": "REPST: AirQuality",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/REPST/AirQuality.yaml"
-        },
-        
-        // AEPSA 模型组
-        {
-            "name": "AEPSA: PEMS-BAY",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/AEPSA/PEMS-BAY.yaml"
-        },
-        {
-            "name": "AEPSA: METR-LA",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/AEPSA/METR-LA.yaml"
-        },
-        {
-            "name": "AEPSA: AirQuality",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/AEPSA/AirQuality.yaml"
-        },
-        {
-            "name": "AEPSA: BJTaxi-Inflow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/AEPSA/BJTaxi-Inflow.yaml"
-        },
-        {
-            "name": "AEPSA: BJTaxi-outflow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/AEPSA/BJTaxi-outflow.yaml"
-        },
-        {
-            "name": "AEPSA: NYCBike-inflow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/AEPSA/NYCBike-inflow.yaml"
-        },
-        {
-            "name": "AEPSA: NYCBike-outflow",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/AEPSA/NYCBike-outflow.yaml"
-        },
-        {
-            "name": "AEPSA: SolarEnergy",
-            "type": "debugpy",
-            "request": "launch",
-            "program": "run.py",
-            "console": "integratedTerminal",
-            "args": "--config ./config/AEPSA/SolarEnergy.yaml"
-        },
-        {
-            "name": "AEPSA_v2: METR-LA",
-            "type": "debugpy",
-            "request": "launch",
-            "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/AGCRN/AirQuality.yaml

@@ -13,7 +13,7 @@ data:
   input_dim: 6
   lag: 24
   normalizer: std
-  num_nodes: 35
+  num_nodes: 12
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 6
   num_layers: 2
+  num_nodes: 12
   output_dim: 6
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 6
   plot: false
   real_value: true
-  seed: 10
+  weight_decay: 0
-  weight_decay: 0

config/AGCRN/BJTaxi-InFlow.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 2
+  num_nodes: 1024
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
+  weight_decay: 0
-  weight_decay: 0

config/AGCRN/BJTaxi-OutFlow.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 2
+  num_nodes: 1024
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
+  weight_decay: 0
-  weight_decay: 0

config/AGCRN/METR-LA.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 2
+  num_nodes: 207
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
+  weight_decay: 0
-  weight_decay: 0

config/AGCRN/NYCBike-InFlow.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 2
+  num_nodes: 128
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
+  weight_decay: 0
-  weight_decay: 0

config/AGCRN/NYCBike-OutFlow.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 2
+  num_nodes: 128
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
+  weight_decay: 0
-  weight_decay: 0

config/AGCRN/PEMSD3.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 2
+  num_nodes: 358
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/AGCRN/PEMSD4.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 2
+  num_nodes: 307
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/AGCRN/PEMSD7.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 2
+  num_nodes: 883
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/AGCRN/PEMSD8.yaml

@@ -24,6 +24,7 @@ model:
   embed_dim: 2
   input_dim: 1
   num_layers: 2
+  num_nodes: 170
   output_dim: 1
   rnn_units: 64
 
@@ -38,13 +39,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 12
   weight_decay: 0

config/AGCRN/SolarEnergy.yaml

@@ -46,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.001
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
+  weight_decay: 0
-  weight_decay: 0

config/ARIMA/AirQuality.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: ARIMA
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -12,17 +13,20 @@ data:
   input_dim: 6
   lag: 24
   normalizer: std
-  num_nodes: 35
+  num_nodes: 12
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   d: 1
   drift: true
   input_dim: 6
+  num_nodes: 12
   output_dim: 6
   p: 2
   q: 1
+
 train:
   batch_size: 16
   debug: false
@@ -36,11 +40,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 6
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/BJTaxi-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: ARIMA
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,13 +17,16 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 1024
   output_dim: 1
   p: 2
   q: 1
+
 train:
   batch_size: 32
   debug: false
@@ -36,11 +40,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/BJTaxi-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: ARIMA
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,13 +17,16 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 1024
   output_dim: 1
   p: 2
   q: 1
+
 train:
   batch_size: 32
   debug: false
@@ -36,11 +40,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/Hainan.yaml

@@ -13,13 +13,14 @@ data:
   input_dim: 1
   lag: 12
   normalizer: std
-  num_nodes: 13
+  num_nodes: 200
   steps_per_day: 288
   test_ratio: 0.2
   val_ratio: 0.2
 
 model:
   input_dim: 1
+  num_nodes: 200
   output_dim: 1
 
 train:
@@ -39,7 +40,7 @@ train:
     - 40
     - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: null
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1

config/ARIMA/METR-LA.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: ARIMA
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -16,13 +17,16 @@ data:
   steps_per_day: 288
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 207
   output_dim: 1
   p: 2
   q: 1
+
 train:
   batch_size: 16
   debug: false
@@ -36,11 +40,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/NYCBike-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: ARIMA
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,17 +13,20 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 128
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 128
   output_dim: 1
   p: 2
   q: 1
+
 train:
   batch_size: 32
   debug: false
@@ -36,11 +40,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/NYCBike-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: ARIMA
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,17 +13,20 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 128
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 128
   output_dim: 1
   p: 2
   q: 1
+
 train:
   batch_size: 32
   debug: false
@@ -36,11 +40,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/PEMSD3.yaml

@@ -22,6 +22,7 @@ model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 358
   output_dim: 1
   p: 2
   q: 1
@@ -39,11 +40,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/PEMSD4.yaml

@@ -22,6 +22,7 @@ model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 307
   output_dim: 1
   p: 2
   q: 1
@@ -37,13 +38,16 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/PEMSD7(L).yaml

@@ -51,5 +51,4 @@ train:
   output_dim: 1
   plot: true
   real_value: true
-  seed: 12
   weight_decay: 0

config/ARIMA/PEMSD7(M).yaml

@@ -51,5 +51,4 @@ train:
   output_dim: 1
   plot: true
   real_value: true
-  seed: 12
   weight_decay: 0

config/ARIMA/PEMSD7.yaml

@@ -22,6 +22,7 @@ model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 883
   output_dim: 1
   p: 2
   q: 1
@@ -49,5 +50,4 @@ train:
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/PEMSD8.yaml

@@ -22,6 +22,7 @@ model:
   d: 1
   drift: true
   input_dim: 1
+  num_nodes: 170
   output_dim: 1
   p: 2
   q: 1
@@ -37,7 +38,11 @@ train:
   loss_func: mae
   lr_decay: false
   lr_decay_rate: 0.3
-  lr_decay_step: [5, 20, 40, 70]
+  lr_decay_step:
+    - 5
+    - 20
+    - 40
+    - 70
   lr_init: 0.003
   mae_thresh: None
   mape_thresh: 0.001
@@ -45,5 +50,4 @@ train:
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ARIMA/SolarEnergy.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: ARIMA
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -16,13 +17,16 @@ data:
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   d: 1
   drift: true
   input_dim: 137
+  num_nodes: 137
   output_dim: 137
   p: 2
   q: 1
+
 train:
   batch_size: 16
   debug: false
@@ -36,11 +40,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 137
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/ASTRA/AirQuality.yaml

@@ -2,7 +2,7 @@ basic:
   dataset: AirQuality
   device: cuda:0
   mode: train
-  model: AEPSA
+  model: ASTRA
   seed: 2023
 
 data:
@@ -32,6 +32,7 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+  output_dim: 6
 
 train:
   batch_size: 16

config/ASTRA/BJTaxi-InFlow.yaml

@@ -2,7 +2,7 @@ basic:
   dataset: BJTaxi-InFlow
   device: cuda:0
   mode: train
-  model: AEPSA
+  model: ASTRA
   seed: 2023
 
 data:
@@ -32,6 +32,7 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+  output_dim: 1
 
 train:
   batch_size: 32

config/ASTRA/BJTaxi-OutFlow.yaml

@@ -2,7 +2,7 @@ basic:
   dataset: BJTaxi-OutFlow
   device: cuda:0
   mode: train
-  model: AEPSA
+  model: ASTRA
   seed: 2023
 
 data:
@@ -17,7 +17,8 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
-
+  output_dim: 1
+  
 model:
   d_ff: 128
   d_model: 64

config/ASTRA/METR-LA.yaml

@@ -2,7 +2,7 @@ basic:
   dataset: METR-LA
   device: cuda:0
   mode: train
-  model: AEPSA
+  model: ASTRA
   seed: 2023
 
 data:
@@ -32,6 +32,7 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+  output_dim: 1
 
 train:
   batch_size: 16

config/ASTRA/NYCBike-InFlow.yaml

@@ -2,7 +2,7 @@ basic:
   dataset: NYCBike-InFlow
   device: cuda:0
   mode: train
-  model: AEPSA
+  model: ASTRA
   seed: 2023
 
 data:
@@ -14,7 +14,7 @@ data:
   lag: 24
   normalizer: std
   num_nodes: 128
-  steps_per_day: 24
+  steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
 
@@ -32,7 +32,8 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
-
+  output_dim: 1
+  
 train:
   batch_size: 32
   debug: false

config/ASTRA/NYCBike-OutFlow.yaml

@@ -2,7 +2,7 @@ basic:
   dataset: NYCBike-OutFlow
   device: cuda:0
   mode: train
-  model: AEPSA
+  model: ASTRA
   seed: 2023
 
 data:
@@ -14,7 +14,7 @@ data:
   lag: 24
   normalizer: std
   num_nodes: 128
-  steps_per_day: 24
+  steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
 
@@ -32,6 +32,7 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+  output_dim: 1
 
 train:
   batch_size: 32

config/ASTRA/PEMS-BAY.yaml

@@ -2,7 +2,7 @@ basic:
   dataset: PEMS-BAY
   device: cuda:0
   mode: train
-  model: AEPSA
+  model: ASTRA
   seed: 2023
 
 data:
@@ -32,6 +32,7 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+  output_dim: 1
 
 train:
   batch_size: 16

config/ASTRA/SolarEnergy.yaml

@@ -2,7 +2,7 @@ basic:
   dataset: SolarEnergy
   device: cuda:0
   mode: train
-  model: AEPSA
+  model: ASTRA
   seed: 2023
 
 data:
@@ -32,6 +32,7 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+  output_dim: 1
 
 train:
   batch_size: 64

config/ASTRA_v2/AirQuality.yaml

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

config/ASTRA_v2/BJTaxi-InFlow.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: BJTaxi-InFlow
+  device: cuda:0
+  mode: train
+  model: ASTRA_v2
+  seed: 2023
+
+data:
+  batch_size: 32
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 1024
+  steps_per_day: 48
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 1024
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 32
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v2/BJTaxi-OutFlow.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: BJTaxi-OutFlow
+  device: cuda:0
+  mode: train
+  model: ASTRA_v2
+  seed: 2023
+
+data:
+  batch_size: 32
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 1024
+  steps_per_day: 48
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 1024
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 32
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v2/METR-LA.yaml

@@ -0,0 +1,59 @@
+basic:
+  dataset: METR-LA
+  device: cuda:0
+  mode: train
+  model: ASTRA_v2
+  seed: 2023
+
+data:
+  batch_size: 16
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 207
+  steps_per_day: 288
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 207
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 16
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 1000
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  real_value: true
+  weight_decay: 0

config/ASTRA_v2/NYCBike-InFlow.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: NYCBike-InFlow
+  device: cuda:0
+  mode: train
+  model: ASTRA_v2
+  seed: 2023
+
+data:
+  batch_size: 32
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 128
+  steps_per_day: 48
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 128
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 32
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v2/NYCBike-OutFlow.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: NYCBike-OutFlow
+  device: cuda:0
+  mode: train
+  model: ASTRA_v2
+  seed: 2023
+
+data:
+  batch_size: 32
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 128
+  steps_per_day: 48
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 128
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 32
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v2/PEMS-BAY.yaml

@@ -1,8 +1,8 @@
 basic:
-  dataset: BJTaxi-InFlow
+  dataset: PEMS-BAY
   device: cuda:0
   mode: train
-  model: REPST
+  model: ASTRA_v2
   seed: 2023
 
 data:
@@ -13,12 +13,13 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 325
-  steps_per_day: 48
+  steps_per_day: 288
   test_ratio: 0.2
   val_ratio: 0.2
 
 model:
+  cheb: 3
   d_ff: 128
   d_model: 64
   dropout: 0.2
@@ -26,12 +27,15 @@ model:
   gpt_path: ./GPT-2
   input_dim: 1
   n_heads: 1
-  num_nodes: 1024
+  num_nodes: 325
   patch_len: 6
   pred_len: 24
   seq_len: 24
   stride: 7
   word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
 
 train:
   batch_size: 16
@@ -51,5 +55,4 @@ train:
   max_grad_norm: 5
   output_dim: 1
   plot: false
-  real_value: true
   weight_decay: 0

config/ASTRA_v2/SolarEnergy.yaml

@@ -2,11 +2,11 @@ basic:
   dataset: SolarEnergy
   device: cuda:0
   mode: train
-  model: AEPSA_v2
+  model: ASTRA_v2
   seed: 2023
 
 data:
-  batch_size: 64
+  batch_size: 16
   column_wise: false
   days_per_week: 7
   horizon: 24
@@ -19,6 +19,7 @@ data:
   val_ratio: 0.2
 
 model:
+  cheb: 3
   d_ff: 128
   d_model: 64
   dropout: 0.2
@@ -32,9 +33,12 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
 
 train:
-  batch_size: 64
+  batch_size: 16
   debug: false
   early_stop: true
   early_stop_patience: 15

config/ASTRA_v3/AirQuality.yaml

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

config/ASTRA_v3/BJTaxi-InFlow.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: BJTaxi-InFlow
+  device: cuda:0
+  mode: train
+  model: ASTRA_v3
+  seed: 2023
+
+data:
+  batch_size: 32
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 1024
+  steps_per_day: 48
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 1024
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 32
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v3/BJTaxi-OutFlow.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: BJTaxi-OutFlow
+  device: cuda:0
+  mode: train
+  model: ASTRA_v3
+  seed: 2023
+
+data:
+  batch_size: 32
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 1024
+  steps_per_day: 48
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 1024
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 32
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v3/METR-LA.yaml

@@ -0,0 +1,59 @@
+basic:
+  dataset: METR-LA
+  device: cuda:0
+  mode: train
+  model: ASTRA_v3
+  seed: 2023
+
+data:
+  batch_size: 16
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 207
+  steps_per_day: 288
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 207
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 16
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 1000
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  real_value: true
+  weight_decay: 0

config/ASTRA_v3/NYCBike-InFlow.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: NYCBike-InFlow
+  device: cuda:0
+  mode: train
+  model: ASTRA_v3
+  seed: 2023
+
+data:
+  batch_size: 32
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 128
+  steps_per_day: 48
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 128
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 32
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v3/NYCBike-OutFlow.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: NYCBike-OutFlow
+  device: cuda:0
+  mode: train
+  model: ASTRA_v3
+  seed: 2023
+
+data:
+  batch_size: 32
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 128
+  steps_per_day: 48
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 128
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 32
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v3/PEMS-BAY.yaml

@@ -0,0 +1,58 @@
+basic:
+  dataset: PEMS-BAY
+  device: cuda:0
+  mode: train
+  model: ASTRA_v3
+  seed: 2023
+
+data:
+  batch_size: 16
+  column_wise: false
+  days_per_week: 7
+  horizon: 24
+  input_dim: 1
+  lag: 24
+  normalizer: std
+  num_nodes: 325
+  steps_per_day: 288
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  cheb: 3
+  d_ff: 128
+  d_model: 64
+  dropout: 0.2
+  gpt_layers: 9
+  gpt_path: ./GPT-2
+  input_dim: 1
+  n_heads: 1
+  num_nodes: 325
+  patch_len: 6
+  pred_len: 24
+  seq_len: 24
+  stride: 7
+  word_num: 1000
+  graph_dim: 64
+  graph_embed_dim: 10
+  output_dim: 1
+
+train:
+  batch_size: 16
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 100
+  grad_norm: false
+  log_step: 100
+  loss_func: mae
+  lr_decay: true
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: None
+  mape_thresh: 0.001
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  weight_decay: 0

config/ASTRA_v3/SolarEnergy.yaml

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

config/DCRNN/AirQuality.yaml

@@ -13,7 +13,7 @@ data:
   input_dim: 6
   lag: 24
   normalizer: std
-  num_nodes: 35
+  num_nodes: 12
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
@@ -25,6 +25,7 @@ model:
   input_dim: 6
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 12
   num_rnn_layers: 2
   output_dim: 6
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.1
   lr_decay_step: 10,20,40,80
   lr_init: 0.001
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 6
   plot: false
   real_value: false
-  seed: 10
+  weight_decay: 0.0001
-  weight_decay: 0.0001

config/DCRNN/BJTaxi-InFlow.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 1024
   num_rnn_layers: 2
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.1
   lr_decay_step: 10,20,40,80
   lr_init: 0.001
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: false
-  seed: 10
+  weight_decay: 0.0001
-  weight_decay: 0.0001

config/DCRNN/BJTaxi-OutFlow.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 1024
   num_rnn_layers: 2
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.1
   lr_decay_step: 10,20,40,80
   lr_init: 0.001
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: false
-  seed: 10
+  weight_decay: 0.0001
-  weight_decay: 0.0001

config/DCRNN/METR-LA.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 207
   num_rnn_layers: 2
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.1
   lr_decay_step: 10,20,40,80
   lr_init: 0.001
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: false
-  seed: 10
+  weight_decay: 0.0001
-  weight_decay: 0.0001

config/DCRNN/NYCBike-InFlow.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 128
   num_rnn_layers: 2
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.1
   lr_decay_step: 10,20,40,80
   lr_init: 0.001
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: false
-  seed: 10
+  weight_decay: 0.0001
-  weight_decay: 0.0001

config/DCRNN/NYCBike-OutFlow.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 128
   num_rnn_layers: 2
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.1
   lr_decay_step: 10,20,40,80
   lr_init: 0.001
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: false
-  seed: 10
+  weight_decay: 0.0001
-  weight_decay: 0.0001

config/DCRNN/PEMSD3.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 358
   num_rnn_layers: 1
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DCRNN/PEMSD4.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 307
   num_rnn_layers: 2
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.1
   lr_decay_step: 10,20,40,80
   lr_init: 0.001
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: false
-  seed: 10
   weight_decay: 0.0001

config/DCRNN/PEMSD7.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 883
   num_rnn_layers: 1
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DCRNN/PEMSD8.yaml

@@ -25,6 +25,7 @@ model:
   input_dim: 1
   l1_decay: 0
   max_diffusion_step: 2
+  num_nodes: 170
   num_rnn_layers: 1
   output_dim: 1
   rnn_units: 64
@@ -44,11 +45,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DCRNN/SolarEnergy.yaml

@@ -45,11 +45,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.001
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
+  weight_decay: 0
-  weight_decay: 0

config/DDGCRN/AirQuality.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DDGCRN
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -12,19 +13,22 @@ data:
   input_dim: 6
   lag: 24
   normalizer: std
-  num_nodes: 35
+  num_nodes: 12
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 12
   input_dim: 6
   num_layers: 1
+  num_nodes: 12
   output_dim: 6
   rnn_units: 64
   use_day: true
   use_week: false
+
 train:
   batch_size: 16
   debug: false
@@ -38,11 +42,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 6
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DDGCRN/BJTaxi-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DDGCRN
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,15 +17,18 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 1024
   output_dim: 1
   rnn_units: 64
   use_day: true
   use_week: false
+
 train:
   batch_size: 32
   debug: false
@@ -38,11 +42,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DDGCRN/BJTaxi-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DDGCRN
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,15 +17,18 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 1024
   output_dim: 1
   rnn_units: 64
   use_day: true
   use_week: false
+
 train:
   batch_size: 32
   debug: false
@@ -38,11 +42,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DDGCRN/Hainan.yaml

@@ -13,7 +13,7 @@ data:
   input_dim: 1
   lag: 12
   normalizer: std
-  num_nodes: 13
+  num_nodes: 200
   steps_per_day: 288
   test_ratio: 0.2
   val_ratio: 0.2
@@ -25,6 +25,7 @@ model:
   horizon: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 200
   output_dim: 1
   rnn_units: 32
   use_day: true
@@ -47,7 +48,7 @@ train:
     - 40
     - 70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: null
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1

config/DDGCRN/METR-LA.yaml

@@ -48,5 +48,4 @@ train:
   max_grad_norm: 5
   output_dim: 1
   plot: false
-  seed: 10
   weight_decay: 0

config/DDGCRN/NYCBike-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DDGCRN
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,19 +13,22 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 128
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 128
   output_dim: 1
   rnn_units: 64
   use_day: true
   use_week: false
+
 train:
   batch_size: 32
   debug: false
@@ -38,11 +42,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DDGCRN/NYCBike-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DDGCRN
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,19 +13,22 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 128
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 128
   output_dim: 1
   rnn_units: 64
   use_day: true
   use_week: false
+
 train:
   batch_size: 32
   debug: false
@@ -38,11 +42,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DDGCRN/PEMSD3.yaml

@@ -23,6 +23,7 @@ model:
   embed_dim: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 358
   output_dim: 1
   rnn_units: 64
   use_day: true
@@ -41,11 +42,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DDGCRN/PEMSD4.yaml

@@ -23,6 +23,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 1
+  num_nodes: 307
   output_dim: 1
   rnn_units: 64
   use_day: true
@@ -41,11 +42,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DDGCRN/PEMSD7(L).yaml

@@ -51,5 +51,4 @@ train:
   output_dim: 1
   plot: true
   real_value: true
-  seed: 12
   weight_decay: 0

config/DDGCRN/PEMSD7(M).yaml

@@ -51,5 +51,4 @@ train:
   output_dim: 1
   plot: true
   real_value: true
-  seed: 12
   weight_decay: 0

config/DDGCRN/PEMSD7.yaml

@@ -23,6 +23,7 @@ model:
   embed_dim: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 883
   output_dim: 1
   rnn_units: 64
   use_day: true
@@ -51,5 +52,4 @@ train:
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DDGCRN/PEMSD8.yaml

@@ -49,5 +49,4 @@ train:
   output_dim: 1
   plot: false
   real_value: true
-  seed: 12
   weight_decay: 0

config/DDGCRN/SolarEnergy.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DDGCRN
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -16,15 +17,18 @@ data:
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 12
   input_dim: 137
   num_layers: 1
+  num_nodes: 137
   output_dim: 137
   rnn_units: 64
   use_day: true
   use_week: false
+
 train:
   batch_size: 16
   debug: false
@@ -38,11 +42,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 137
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/AirQuality.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DSANET
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -12,10 +13,11 @@ data:
   input_dim: 6
   lag: 24
   normalizer: std
-  num_nodes: 35
+  num_nodes: 12
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   batch_size: 64
   d_inner: 2048
@@ -29,9 +31,11 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 35
+  num_nodes: 12
   output_dim: 6
   w_kernel: 1
   window: 24
+
 train:
   batch_size: 16
   debug: false
@@ -45,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 6
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/BJTaxi-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DSANET
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,6 +17,7 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   batch_size: 64
   d_inner: 2048
@@ -29,9 +31,11 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 1024
+  num_nodes: 1024
   output_dim: 1
   w_kernel: 1
   window: 24
+
 train:
   batch_size: 32
   debug: false
@@ -45,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/BJTaxi-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DSANET
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,6 +17,7 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   batch_size: 64
   d_inner: 2048
@@ -29,9 +31,11 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 1024
+  num_nodes: 1024
   output_dim: 1
   w_kernel: 1
   window: 24
+
 train:
   batch_size: 32
   debug: false
@@ -45,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/METR-LA.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DSANET
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -16,6 +17,7 @@ data:
   steps_per_day: 288
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   batch_size: 64
   d_inner: 2048
@@ -29,9 +31,11 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 207
+  num_nodes: 207
   output_dim: 1
   w_kernel: 1
   window: 12
+
 train:
   batch_size: 16
   debug: false
@@ -45,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/NYCBike-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DSANET
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,10 +13,11 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 128
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   batch_size: 64
   d_inner: 2048
@@ -29,9 +31,11 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 1024
+  num_nodes: 128
   output_dim: 1
   w_kernel: 1
   window: 24
+
 train:
   batch_size: 32
   debug: false
@@ -45,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/NYCBike-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DSANET
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,10 +13,11 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 128
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   batch_size: 64
   d_inner: 2048
@@ -29,9 +31,11 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 1024
+  num_nodes: 128
   output_dim: 1
   w_kernel: 1
   window: 24
+
 train:
   batch_size: 32
   debug: false
@@ -45,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/PEMSD3.yaml

@@ -31,6 +31,7 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 358
+  num_nodes: 358
   output_dim: 1
   w_kernel: 1
   window: 12
@@ -48,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/PEMSD4.yaml

@@ -31,6 +31,7 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 307
+  num_nodes: 307
   output_dim: 1
   w_kernel: 1
   window: 12
@@ -48,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/PEMSD7.yaml

@@ -31,6 +31,7 @@ model:
   n_kernels: 32
   n_layers: 3
   n_multiv: 883
+  num_nodes: 883
   output_dim: 1
   w_kernel: 1
   window: 12
@@ -48,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/PEMSD8.yaml

@@ -31,6 +31,7 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 170
+  num_nodes: 170
   output_dim: 1
   w_kernel: 1
   window: 12
@@ -48,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/DSANET/SolarEnergy.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: DSANET
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -16,6 +17,7 @@ data:
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   batch_size: 64
   d_inner: 2048
@@ -29,9 +31,11 @@ model:
   n_kernels: 32
   n_layers: 6
   n_multiv: 137
+  num_nodes: 137
   output_dim: 137
   w_kernel: 1
   window: 24
+
 train:
   batch_size: 16
   debug: false
@@ -45,11 +49,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 137
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/AirQuality.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXP
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -12,10 +13,11 @@ data:
   input_dim: 6
   lag: 24
   normalizer: std
-  num_nodes: 35
+  num_nodes: 12
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
@@ -24,11 +26,13 @@ model:
   in_len: 24
   input_dim: 6
   num_layers: 1
+  num_nodes: 12
   output_dim: 6
   rnn_units: 64
   top_k: 2
   use_day: true
   use_week: true
+
 train:
   batch_size: 16
   debug: false
@@ -42,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 6
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/BJTaxi-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXP
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,6 +17,7 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
@@ -24,11 +26,13 @@ model:
   in_len: 24
   input_dim: 1
   num_layers: 1
+  num_nodes: 1024
   output_dim: 1
   rnn_units: 64
   top_k: 2
   use_day: true
   use_week: true
+
 train:
   batch_size: 32
   debug: false
@@ -42,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/BJTaxi-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXP
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,6 +17,7 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
@@ -24,11 +26,13 @@ model:
   in_len: 24
   input_dim: 1
   num_layers: 1
+  num_nodes: 1024
   output_dim: 1
   rnn_units: 64
   top_k: 2
   use_day: true
   use_week: true
+
 train:
   batch_size: 32
   debug: false
@@ -42,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/METR-LA.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXP
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -16,6 +17,7 @@ data:
   steps_per_day: 288
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
@@ -24,11 +26,13 @@ model:
   in_len: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 207
   output_dim: 1
   rnn_units: 64
   top_k: 2
   use_day: true
   use_week: true
+
 train:
   batch_size: 16
   debug: false
@@ -42,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/NYCBike-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXP
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,10 +13,11 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 128
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
@@ -24,11 +26,13 @@ model:
   in_len: 24
   input_dim: 1
   num_layers: 1
+  num_nodes: 128
   output_dim: 1
   rnn_units: 64
   top_k: 2
   use_day: true
   use_week: true
+
 train:
   batch_size: 32
   debug: false
@@ -42,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/NYCBike-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXP
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,10 +13,11 @@ data:
   input_dim: 1
   lag: 24
   normalizer: std
-  num_nodes: 1024
+  num_nodes: 128
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
@@ -24,11 +26,13 @@ model:
   in_len: 24
   input_dim: 1
   num_layers: 1
+  num_nodes: 128
   output_dim: 1
   rnn_units: 64
   top_k: 2
   use_day: true
   use_week: true
+
 train:
   batch_size: 32
   debug: false
@@ -42,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/PEMSD3.yaml

@@ -26,6 +26,7 @@ model:
   in_len: 12
   input_dim: 1
   num_layers: 1
+  num_nodes: 358
   output_dim: 1
   rnn_units: 64
   top_k: 2
@@ -45,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/PEMSD4.yaml

@@ -23,6 +23,7 @@ model:
   cycle_len: 288
   in_len: 12
   input_dim: 1
+  num_nodes: 307
   output_dim: 1
 
 train:
@@ -38,11 +39,10 @@ train:
   lr_decay_rate: 0.5
   lr_decay_step: 5,20,40,65
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/PEMSD7.yaml

@@ -22,6 +22,7 @@ model:
   batch_size: 64
   in_len: 12
   input_dim: 1
+  num_nodes: 883
   output_dim: 1
 
 train:
@@ -37,11 +38,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/PEMSD8.yaml

@@ -43,11 +43,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXP/SD.yaml

@@ -13,7 +13,7 @@ data:
   input_dim: 1
   lag: 12
   normalizer: std
-  num_nodes: 716
+  num_nodes: 307
   steps_per_day: 288
   test_ratio: 0.2
   val_ratio: 0.2
@@ -22,6 +22,7 @@ model:
   batch_size: 64
   in_len: 12
   input_dim: 1
+  num_nodes: 307
   output_dim: 1
 
 train:
@@ -37,7 +38,7 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh:
+  mae_thresh: null
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1

config/EXP/SolarEnergy.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXP
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -16,6 +17,7 @@ data:
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
@@ -24,11 +26,13 @@ model:
   in_len: 24
   input_dim: 137
   num_layers: 1
+  num_nodes: 137
   output_dim: 137
   rnn_units: 64
   top_k: 2
   use_day: true
   use_week: true
+
 train:
   batch_size: 16
   debug: false
@@ -42,11 +46,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 137
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXPB/AirQuality.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXPB
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -12,20 +13,23 @@ data:
   input_dim: 6
   lag: 24
   normalizer: std
-  num_nodes: 35
+  num_nodes: 12
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
   input_dim: 6
   num_layers: 1
+  num_nodes: 12
   output_dim: 6
   patch_size: 3
   rnn_units: 64
   use_day: true
   use_week: true
+
 train:
   batch_size: 16
   debug: false
@@ -39,11 +43,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 6
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/EXPB/BJTaxi-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXPB
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -16,16 +17,19 @@ data:
   steps_per_day: 48
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
   input_dim: 1
   num_layers: 1
+  num_nodes: 1024
   output_dim: 1
   patch_size: 3
   rnn_units: 64
   use_day: true
   use_week: true
+
 train:
   batch_size: 32
   debug: false
@@ -39,11 +43,10 @@ train:
   lr_decay_rate: 0.3
   lr_decay_step: 5,20,40,70
   lr_init: 0.003
-  mae_thresh: ''
+  mae_thresh: 0.0
   mape_thresh: 0.0
   max_grad_norm: 5
   output_dim: 1
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0