Merge pull request 'REPST' (#3 ) from REPST into main

Reviewed-on: #3
impl STNorm
2025-12-20 16:03:22 +08:00 · 2025-12-20 15:45:13 +08:00 · 2025-12-19 11:54:07 +08:00 · 2025-12-19 10:19:17 +08:00 · 2025-12-17 17:13:16 +08:00 · 2025-12-16 21:47:17 +08:00
465 changed files with 10747 additions and 3684 deletions
--- a/.vscode/launch.json
+++ b/.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"
-        },
-        // 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"
-        },
+            "justMyCode": false
+        }
    ]
 }
--- a/config/AGCRN/AirQuality.yaml
+++ b/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
--- a/config/AGCRN/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/AGCRN/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/AGCRN/METR-LA.yaml
+++ b/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
--- a/config/AGCRN/NYCBike-InFlow.yaml
+++ b/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
--- a/config/AGCRN/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/AGCRN/PEMSD3.yaml
+++ b/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
--- a/config/AGCRN/PEMSD4.yaml
+++ b/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
--- a/config/AGCRN/PEMSD7.yaml
+++ b/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
--- a/config/AGCRN/PEMSD8.yaml
+++ b/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
--- a/config/AGCRN/SolarEnergy.yaml
+++ b/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
--- a/config/ARIMA/AirQuality.yaml
+++ b/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
--- a/config/ARIMA/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/ARIMA/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/ARIMA/Hainan.yaml
+++ b/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
--- a/config/ARIMA/METR-LA.yaml
+++ b/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
--- a/config/ARIMA/NYCBike-InFlow.yaml
+++ b/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
--- a/config/ARIMA/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/ARIMA/PEMSD3.yaml
+++ b/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
--- a/config/ARIMA/PEMSD4.yaml
+++ b/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
--- a/config/ARIMA/PEMSD7(L).yaml
+++ b/config/ARIMA/PEMSD7(L).yaml
@ -51,5 +51,4 @@ train:
  output_dim: 1
  plot: true
  real_value: true
-  seed: 12
  weight_decay: 0
--- a/config/ARIMA/PEMSD7(M).yaml
+++ b/config/ARIMA/PEMSD7(M).yaml
@ -51,5 +51,4 @@ train:
  output_dim: 1
  plot: true
  real_value: true
-  seed: 12
  weight_decay: 0
--- a/config/ARIMA/PEMSD7.yaml
+++ b/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
--- a/config/ARIMA/PEMSD8.yaml
+++ b/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
--- a/config/ARIMA/SolarEnergy.yaml
+++ b/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
--- a/config/ASTRA/AirQuality.yaml
+++ b/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
--- a/config/ASTRA/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/ASTRA/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/ASTRA/METR-LA.yaml
+++ b/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
--- a/config/ASTRA/NYCBike-InFlow.yaml
+++ b/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
--- a/config/ASTRA/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/ASTRA/PEMS-BAY.yaml
+++ b/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
--- a/config/ASTRA/SolarEnergy.yaml
+++ b/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
--- a/config/ASTRA_v2/AirQuality.yaml
+++ b/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
--- a/config/ASTRA_v2/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/ASTRA_v2/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/ASTRA_v2/METR-LA.yaml
+++ b/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
--- a/config/ASTRA_v2/NYCBike-InFlow.yaml
+++ b/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
--- a/config/ASTRA_v2/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/REPST/BJTaxi-Inflow.yaml
+++ b/config/REPST/BJTaxi-Inflow.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
-  steps_per_day: 48
+  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
@ -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
--- a/config/ASTRA_v2/SolarEnergy.yaml
+++ b/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
--- a/config/ASTRA_v3/AirQuality.yaml
+++ b/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
--- a/config/ASTRA_v3/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/ASTRA_v3/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/ASTRA_v3/METR-LA.yaml
+++ b/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
--- a/config/ASTRA_v3/NYCBike-InFlow.yaml
+++ b/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
--- a/config/ASTRA_v3/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/ASTRA_v3/PEMS-BAY.yaml
+++ b/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
--- a/config/ASTRA_v3/SolarEnergy.yaml
+++ b/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
--- a/config/DCRNN/AirQuality.yaml
+++ b/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
--- a/config/DCRNN/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/DCRNN/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/DCRNN/METR-LA.yaml
+++ b/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
--- a/config/DCRNN/NYCBike-InFlow.yaml
+++ b/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
--- a/config/DCRNN/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/DCRNN/PEMSD3.yaml
+++ b/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
--- a/config/DCRNN/PEMSD4.yaml
+++ b/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
--- a/config/DCRNN/PEMSD7.yaml
+++ b/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
--- a/config/DCRNN/PEMSD8.yaml
+++ b/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
--- a/config/DCRNN/SolarEnergy.yaml
+++ b/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
--- a/config/DDGCRN/AirQuality.yaml
+++ b/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
--- a/config/DDGCRN/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/DDGCRN/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/DDGCRN/Hainan.yaml
+++ b/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
--- a/config/DDGCRN/METR-LA.yaml
+++ b/config/DDGCRN/METR-LA.yaml
@ -48,5 +48,4 @@ train:
  max_grad_norm: 5
  output_dim: 1
  plot: false
-  seed: 10
  weight_decay: 0
--- a/config/DDGCRN/NYCBike-InFlow.yaml
+++ b/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
--- a/config/DDGCRN/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/DDGCRN/PEMSD3.yaml
+++ b/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
--- a/config/DDGCRN/PEMSD4.yaml
+++ b/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
--- a/config/DDGCRN/PEMSD7(L).yaml
+++ b/config/DDGCRN/PEMSD7(L).yaml
@ -51,5 +51,4 @@ train:
  output_dim: 1
  plot: true
  real_value: true
-  seed: 12
  weight_decay: 0
--- a/config/DDGCRN/PEMSD7(M).yaml
+++ b/config/DDGCRN/PEMSD7(M).yaml
@ -51,5 +51,4 @@ train:
  output_dim: 1
  plot: true
  real_value: true
-  seed: 12
  weight_decay: 0
--- a/config/DDGCRN/PEMSD7.yaml
+++ b/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
--- a/config/DDGCRN/PEMSD8.yaml
+++ b/config/DDGCRN/PEMSD8.yaml
@ -49,5 +49,4 @@ train:
  output_dim: 1
  plot: false
  real_value: true
-  seed: 12
  weight_decay: 0
--- a/config/DDGCRN/SolarEnergy.yaml
+++ b/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
--- a/config/DSANET/AirQuality.yaml
+++ b/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
--- a/config/DSANET/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/DSANET/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/DSANET/METR-LA.yaml
+++ b/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
--- a/config/DSANET/NYCBike-InFlow.yaml
+++ b/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
--- a/config/DSANET/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/DSANET/PEMSD3.yaml
+++ b/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
--- a/config/DSANET/PEMSD4.yaml
+++ b/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
--- a/config/DSANET/PEMSD7.yaml
+++ b/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
--- a/config/DSANET/PEMSD8.yaml
+++ b/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
--- a/config/DSANET/SolarEnergy.yaml
+++ b/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
--- a/config/EXP/AirQuality.yaml
+++ b/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
--- a/config/EXP/BJTaxi-InFlow.yaml
+++ b/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
--- a/config/EXP/BJTaxi-OutFlow.yaml
+++ b/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
--- a/config/EXP/METR-LA.yaml
+++ b/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
--- a/config/EXP/NYCBike-InFlow.yaml
+++ b/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
--- a/config/EXP/NYCBike-OutFlow.yaml
+++ b/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
--- a/config/EXP/PEMSD3.yaml
+++ b/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
--- a/config/EXP/PEMSD4.yaml
+++ b/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
--- a/config/EXP/PEMSD7.yaml
+++ b/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
--- a/config/EXP/PEMSD8.yaml
+++ b/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
--- a/config/EXP/SD.yaml
+++ b/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
--- a/config/EXP/SolarEnergy.yaml
+++ b/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
--- a/config/EXPB/AirQuality.yaml
+++ b/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
--- a/config/EXPB/BJTaxi-InFlow.yaml
+++ b/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
--- a/Show More
+++ b/Show More
Author	SHA1	Message	Date
czzhangheng	5a7ec07a12	Merge pull request 'REPST' (#3 ) from REPST into main Reviewed-on: #3	2025-12-20 16:03:22 +08:00
czzhangheng	9d3293cef7	impl STNorm	2025-12-20 15:45:13 +08:00
czzhangheng	6a8d1f33f9	fix repst:Air bug	2025-12-19 11:54:07 +08:00
czzhangheng	cc63e5078f	impl FPT	2025-12-19 10:19:17 +08:00
czzhangheng	21bc05e763	修复TSLoader Bug	2025-12-17 17:13:16 +08:00
czzhangheng	1a13a32688	修复配置。ASTRA v2 or v3使用硬参数，确保配置文件完整	2025-12-16 21:47:17 +08:00
czzhangheng	b97111f5ea	统一REPST:AirQuality配置	2025-12-16 17:38:37 +08:00
czzhangheng	b38e4a5da2	修复astra bug	2025-12-16 17:31:57 +08:00
czzhangheng	85257bc61c	refactor(trainer): 优化训练器代码结构并添加进度条显示调整训练器代码结构，减少冗余代码，提高可读性为训练过程添加tqdm进度条，实时显示loss信息统一TRY环境变量的输出格式简化日志记录和模型保存逻辑	2025-12-16 16:40:40 +08:00
czzhangheng	659b41f612	refactor(config): 统一配置文件命名并调整模型参数更新多个模型的配置文件命名格式，统一使用大驼峰格式调整SolarEnergy和BJTaxi数据集的输入维度和批量大小删除旧命名格式的配置文件并添加新的配置文件修改训练脚本中的模型和数据集列表用于调试	2025-12-15 21:33:28 +08:00
czzhangheng	5e52f23c8d	fix(config): 修正配置文件命名不一致问题并更新训练参数 refactor(trainer): 重构训练器代码，优化反归一化处理和形状转换逻辑 style(trainer): 简化代码格式，提高可读性 chore: 更新训练脚本中的模型和数据集列表	2025-12-15 21:23:04 +08:00
czzhangheng	b6d4f5daf5	refactor(dataloader): 重构数据加载器代码，优化滑动窗口生成和归一化处理重构PeMSD和EXP数据加载器，使用numpy的stride_tricks实现高效滑动窗口统一数据预处理流程，简化代码结构并提高可维护性优化归一化处理，支持多scaler通道独立处理	2025-12-15 20:54:20 +08:00
czzhangheng	56b09ea8ac	feat: 重构训练器和配置结构，优化模型训练流程 refactor(trainer): 重构Trainer类结构，拆分初始化方法 perf(trainer): 优化训练循环和评估逻辑 style(config): 统一配置文件命名和结构 fix(trainer): 修复形状检查逻辑和调试模式处理 docs: 更新README和注释说明	2025-12-15 19:55:57 +08:00
czzhangheng	3095b7435b	refactor: 重构数据加载器和训练器代码，优化代码结构和可读性重构数据加载器模块，使用字典映射替代switch-case结构简化训练器逻辑，合并重复代码，提高可维护性优化日志时间格式，缩短显示长度调整训练配置，减少默认epoch数并启用GPU训练统一数据加载方式，提取公共方法减少重复代码	2025-12-15 01:38:47 +08:00
czzhangheng	97743dfd05	opt trainer	2025-12-15 00:38:43 +08:00
czzhangheng	3b4acd4951	合并	2025-12-14 17:50:08 +08:00
czzhangheng	5827554c73	改进模型注册，动态注册	2025-12-14 17:48:37 +08:00
czzhangheng	9147803c2b	改进trainer，使用统一的loader而不是TSLoader，仅在Trainer上做了shape变换，确定尽可能少改动数据。	2025-12-14 17:47:38 +08:00
czzhangheng	19fd7622a3	兼容InFormer	2025-12-11 23:16:25 +08:00
czzhangheng	600420e8df	impl mtgnn	2025-12-10 23:31:17 +08:00
czzhangheng	4ccb029d7e	impl PatchTST	2025-12-10 21:53:46 +08:00
czzhangheng	d8f4cc5825	简化trainer，修复设备bug，新增批量运行脚本	2025-12-10 21:08:20 +08:00
czzhangheng	44ffe94c95	更新iTransformer, HI配置。更新TS数据集载入方式	2025-12-10 11:01:00 +08:00
czzhangheng	560d24e5a8	更新v2配置	2025-12-10 10:39:41 +08:00
czzhangheng	5c2380ae21	适配HI	2025-12-09 17:49:18 +08:00
czzhangheng	b57fcef039	新增SolarEnergy-iTransformer配置	2025-12-09 16:56:30 +08:00
czzhangheng	faeb90e734	实现iTransformer	2025-12-09 16:12:05 +08:00
czzhangheng	4984d24506	实现iTransformer	2025-12-09 16:11:49 +08:00
czzhangheng	e4a7884c98	更名ASTRA配置	2025-12-09 14:08:59 +08:00
czzhangheng	9c76975056	更名ASTRA	2025-12-09 14:07:38 +08:00
czzhangheng	aed1e53f0f	更名ASTRA	2025-12-09 14:06:56 +08:00
czzhangheng	1598918112	更改v3_PEWSBAY配置	2025-12-06 19:49:31 +08:00
czzhangheng	f899f50b16	变量名更改	2025-12-06 19:48:27 +08:00
czzhangheng	865c5a3082	ASTRA-V3	2025-12-06 19:47:33 +08:00
czzhangheng	07d7d43857	更新配置，METR-LA -STAEFormer	2025-12-03 18:51:54 +08:00
czzhangheng	9b1cf5f0ce	更新测试脚本	2025-12-03 18:11:06 +08:00
czzhangheng	440cb6936b	兼容STAEFormer	2025-12-03 17:12:46 +08:00
czzhangheng	a9313390ac	适配GraphWaveNet	2025-12-03 12:05:02 +08:00
czzhangheng	140ead3975	为model添加num_nodes	2025-12-03 08:40:00 +08:00
czzhangheng	e50f443470	add astra-pemsbay v2	2025-12-02 09:40:24 +08:00
czzhangheng	c4414dd5d9	fix twdgcn config	2025-12-02 09:12:09 +08:00
czzhangheng	77a3210475	测试所有配置是否正常运行	2025-12-01 22:29:52 +08:00
czzhangheng	d4ee8e309e	trainer修改	2025-12-01 21:36:37 +08:00