更新配置，METR-LA -STAEFormer

config/AEPSA/BJTaxi-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: AEPSA
   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:
   d_ff: 128
   d_model: 64
@@ -30,6 +32,7 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+
 train:
   batch_size: 32
   debug: false

config/AEPSA/BJTaxi-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: AEPSA
   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:
   d_ff: 128
   d_model: 64
@@ -30,6 +32,7 @@ model:
   seq_len: 24
   stride: 7
   word_num: 1000
+
 train:
   batch_size: 32
   debug: false

config/AEPSA/NYCBike-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: AEPSA
   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:
   d_ff: 128
   d_model: 64
@@ -24,12 +26,13 @@ model:
   gpt_path: ./GPT-2
   input_dim: 1
   n_heads: 1
-  num_nodes: 1024
+  num_nodes: 128
   patch_len: 6
   pred_len: 24
   seq_len: 24
   stride: 7
   word_num: 1000
+
 train:
   batch_size: 32
   debug: false

config/AEPSA/NYCBike-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: AEPSA
   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:
   d_ff: 128
   d_model: 64
@@ -24,12 +26,13 @@ model:
   gpt_path: ./GPT-2
   input_dim: 1
   n_heads: 1
-  num_nodes: 1024
+  num_nodes: 128
   patch_len: 6
   pred_len: 24
   seq_len: 24
   stride: 7
   word_num: 1000
+
 train:
   batch_size: 32
   debug: false

config/AEPSA/v2_PEMS-BAY.yaml

@@ -0,0 +1,54 @@
+basic:
+  dataset: PEMS-BAY
+  device: cuda:0
+  mode: train
+  model: AEPSA_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: 325
+  steps_per_day: 288
+  test_ratio: 0.2
+  val_ratio: 0.2
+
+model:
+  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
+
+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/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/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

config/EXPB/BJTaxi-OutFlow.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

config/EXPB/METR-LA.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXPB
   seed: 2023
+
 data:
   batch_size: 64
   column_wise: false
@@ -16,16 +17,19 @@ data:
   steps_per_day: 288
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
   input_dim: 1
   num_layers: 1
+  num_nodes: 207
   output_dim: 1
   patch_size: 3
   rnn_units: 64
   use_day: true
   use_week: true
+
 train:
   batch_size: 64
   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

config/EXPB/NYCBike-InFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXPB
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,20 +13,23 @@ 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
   input_dim: 1
   num_layers: 1
+  num_nodes: 128
   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

config/EXPB/NYCBike-OutFlow.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXPB
   seed: 2023
+
 data:
   batch_size: 32
   column_wise: false
@@ -12,20 +13,23 @@ 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
   input_dim: 1
   num_layers: 1
+  num_nodes: 128
   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

config/EXPB/PEMSD4.yaml

@@ -23,6 +23,7 @@ model:
   embed_dim: 10
   input_dim: 1
   num_layers: 1
+  num_nodes: 307
   output_dim: 1
   patch_size: 3
   rnn_units: 64
@@ -42,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/EXPB/SolarEnergy.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: EXPB
   seed: 2023
+
 data:
   batch_size: 16
   column_wise: false
@@ -16,16 +17,19 @@ data:
   steps_per_day: 24
   test_ratio: 0.2
   val_ratio: 0.2
+
 model:
   cheb_order: 2
   embed_dim: 10
   input_dim: 137
   num_layers: 1
+  num_nodes: 137
   output_dim: 137
   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: 137
   plot: false
   real_value: true
-  seed: 10
   weight_decay: 0

config/GWN/AirQuality.yaml

@@ -6,7 +6,7 @@ basic:
   seed: 2023
 
 data:
-  batch_size: 16
+  batch_size: 64
   column_wise: false
   days_per_week: 7
   horizon: 24
@@ -20,26 +20,27 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
-  batch_size: 16
+  apt_size: 10
+  aptinit: null
+  batch_size: 64
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
+  do_graph_conv: True
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
-  input_dim: 6
+  in_dim: 1
+  input_dim: 1
   kernel_size: 2
-  layers: 2
-  out_dim: 12
-  output_dim: 6
+  layers: 4
+  num_nodes: 35
+  out_dim: 24
   residual_channels: 32
   skip_channels: 256
-  supports:
-
+  supports: null
 
 train:
-  batch_size: 16
+  batch_size: 64
   debug: false
   early_stop: true
   early_stop_patience: 15
@@ -51,11 +52,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
+  output_dim: 1
   plot: false
   real_value: true
-  seed: 10
-  weight_decay: 0
+  weight_decay: 0

config/GWN/BJTaxi-InFlow.yaml

@@ -20,22 +20,24 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
-  batch_size: 32
+  apt_size: 10
+  aptinit: null
+  batch_size: 16
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
+  do_graph_conv: True
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 1
   input_dim: 1
   kernel_size: 2
-  layers: 2
-  out_dim: 12
-  output_dim: 1
+  layers: 4
+  num_nodes: 1024
+  out_dim: 24
   residual_channels: 32
   skip_channels: 256
-  supports:
+  supports: null
 
 
 train:
@@ -51,11 +53,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
+  weight_decay: 0

config/GWN/BJTaxi-OutFlow.yaml

@@ -20,23 +20,24 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
+  apt_size: 10
+  aptinit: null
   batch_size: 32
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
+  do_graph_conv: True
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 1
   input_dim: 1
   kernel_size: 2
-  layers: 2
-  out_dim: 12
-  output_dim: 1
+  layers: 4
+  num_nodes: 1024
+  out_dim: 24
   residual_channels: 32
   skip_channels: 256
-  supports:
-
+  supports: null
 
 train:
   batch_size: 32
@@ -51,11 +52,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
+  weight_decay: 0

config/GWN/METR-LA.yaml

@@ -6,7 +6,7 @@ basic:
   seed: 2023
 
 data:
-  batch_size: 16
+  batch_size: 64
   column_wise: false
   days_per_week: 7
   horizon: 24
@@ -20,26 +20,27 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
-  batch_size: 16
+  apt_size: 10
+  aptinit: null
+  batch_size: 64
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
+  do_graph_conv: True
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 1
   input_dim: 1
   kernel_size: 2
-  layers: 2
-  out_dim: 12
-  output_dim: 1
+  layers: 4
+  num_nodes: 207
+  out_dim: 24
   residual_channels: 32
   skip_channels: 256
-  supports:
-
+  supports: null
 
 train:
-  batch_size: 16
+  batch_size: 64
   debug: false
   early_stop: true
   early_stop_patience: 15
@@ -51,11 +52,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
+  weight_decay: 0

config/GWN/NYCBike-InFlow.yaml

@@ -20,23 +20,24 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
+  apt_size: 10
+  aptinit: null
   batch_size: 32
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
+  do_graph_conv: True
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 1
   input_dim: 1
   kernel_size: 2
-  layers: 2
-  out_dim: 12
-  output_dim: 1
+  layers: 4
+  num_nodes: 128
+  out_dim: 24
   residual_channels: 32
   skip_channels: 256
-  supports:
-
+  supports: null
 
 train:
   batch_size: 32
@@ -51,11 +52,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
+  weight_decay: 0

config/GWN/NYCBike-OutFlow.yaml

@@ -6,7 +6,7 @@ basic:
   seed: 2023
 
 data:
-  batch_size: 32
+  batch_size: 16
   column_wise: false
   days_per_week: 7
   horizon: 24
@@ -20,26 +20,27 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
-  batch_size: 32
+  apt_size: 10
+  aptinit: null
+  batch_size: 16
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
+  do_graph_conv: True
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 1
   input_dim: 1
   kernel_size: 2
-  layers: 2
-  out_dim: 12
-  output_dim: 1
+  layers: 4
+  num_nodes: 128
+  out_dim: 24
   residual_channels: 32
   skip_channels: 256
-  supports:
-
+  supports: null
 
 train:
-  batch_size: 32
+  batch_size: 16
   debug: false
   early_stop: true
   early_stop_patience: 15
@@ -51,11 +52,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
+  weight_decay: 0

config/GWN/PEMS-BAY.yaml

@@ -0,0 +1,61 @@
+basic:
+  dataset: PEMS-BAY
+  device: cuda:0
+  mode: train
+  model: GWN
+  seed: 2023
+
+data:
+  batch_size: 64
+  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:
+  addaptadj: true
+  apt_size: 10
+  aptinit: null
+  batch_size: 64
+  blocks: 4
+  dilation_channels: 32
+  dropout: 0.3
+  do_graph_conv: True
+  end_channels: 512
+  gcn_bool: true
+  in_dim: 1
+  input_dim: 1
+  kernel_size: 2
+  layers: 4
+  num_nodes: 325
+  out_dim: 24
+  residual_channels: 32
+  skip_channels: 256
+  supports: null
+
+train:
+  batch_size: 64
+  debug: false
+  early_stop: true
+  early_stop_patience: 15
+  epochs: 300
+  grad_norm: false
+  log_step: 1000
+  loss_func: mae
+  lr_decay: false
+  lr_decay_rate: 0.3
+  lr_decay_step: 5,20,40,70
+  lr_init: 0.003
+  mae_thresh: 0.0
+  mape_thresh: 0.0
+  max_grad_norm: 5
+  output_dim: 1
+  plot: false
+  real_value: true
+  weight_decay: 0

config/GWN/PEMSD3.yaml

@@ -20,30 +20,23 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
+  aptinit: null
   batch_size: 64
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 3
   input_dim: 1
   kernel_size: 2
   layers: 2
+  num_nodes: 358
   out_dim: 12
   output_dim: 1
   residual_channels: 32
   skip_channels: 256
-  supports:
-
-
-
-
-
-
-
-
+  supports: null
 
 train:
   batch_size: 16
@@ -58,11 +51,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/GWN/PEMSD4.yaml

@@ -20,30 +20,23 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
+  aptinit: null
   batch_size: 64
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 1
   input_dim: 1
   kernel_size: 2
   layers: 2
+  num_nodes: 307
   out_dim: 12
   output_dim: 1
   residual_channels: 32
   skip_channels: 256
-  supports:
-
-
-
-
-
-
-
-
+  supports: null
 
 train:
   batch_size: 64
@@ -58,11 +51,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/GWN/PEMSD7.yaml

@@ -20,30 +20,23 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
+  aptinit: null
   batch_size: 64
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 3
   input_dim: 1
   kernel_size: 2
   layers: 2
+  num_nodes: 883
   out_dim: 12
   output_dim: 1
   residual_channels: 32
   skip_channels: 256
-  supports:
-
-
-
-
-
-
-
-
+  supports: null
 
 train:
   batch_size: 16
@@ -58,11 +51,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/GWN/PEMSD8.yaml

@@ -20,30 +20,23 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
+  aptinit: null
   batch_size: 64
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 3
   input_dim: 1
   kernel_size: 2
   layers: 2
+  num_nodes: 170
   out_dim: 12
   output_dim: 1
   residual_channels: 32
   skip_channels: 256
-  supports:
-
-
-
-
-
-
-
-
+  supports: null
 
 train:
   batch_size: 64
@@ -58,11 +51,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/GWN/SolarEnergy.yaml

@@ -20,22 +20,24 @@ data:
 
 model:
   addaptadj: true
-  aptinit:
-  batch_size: 64
+  apt_size: 10
+  aptinit: null
+  batch_size: 32
   blocks: 4
   dilation_channels: 32
   dropout: 0.3
+  do_graph_conv: True
   end_channels: 512
   gcn_bool: true
-  in_dim: 2
+  in_dim: 1
   input_dim: 1
   kernel_size: 2
-  layers: 2
-  out_dim: 12
-  output_dim: 1
+  layers: 4
+  num_nodes: 137
+  out_dim: 24
   residual_channels: 32
   skip_channels: 256
-  supports:
+  supports: null
 
 train:
   batch_size: 64
@@ -50,11 +52,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/MegaCRN/AirQuality.yaml

@@ -4,6 +4,7 @@ basic:
   mode: train
   model: MegaCRN
   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_k: 3
   cl_decay_steps: 2000
@@ -23,10 +25,12 @@ model:
   mem_dim: 64
   mem_num: 20
   num_layers: 1
+  num_nodes: 12
   output_dim: 6
   rnn_units: 64
   use_curriculum_learning: true
   ycov_dim: 1
+
 train:
   batch_size: 16
   debug: false
@@ -40,11 +44,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