更新配置

config/AEPSA/AirQuality.yaml

@@ -36,6 +36,7 @@ model:
   n_heads: 1
   input_dim: 6
   word_num: 1000
+  num_nodes: 35
 
 train:
   batch_size: 16

config/AEPSA/BJTaxi-Inflow.yaml

@@ -11,10 +11,10 @@ data:
   column_wise: false
   days_per_week: 7
   default_graph: true
-  horizon: 12
-  lag: 12
+  horizon: 24
+  lag: 24
   normalizer: std
-  num_nodes: 142
+  num_nodes: 1024
   steps_per_day: 48
   test_ratio: 0.2
   tod: false
@@ -24,8 +24,8 @@ data:
   batch_size: 32
 
 model:
-  pred_len: 12
-  seq_len: 12
+  pred_len: 24
+  seq_len: 24
   patch_len: 6 
   stride: 7
   dropout: 0.2
@@ -36,6 +36,7 @@ model:
   n_heads: 1
   input_dim: 1
   word_num: 1000
+  num_nodes: 1024
 
 train:
   batch_size: 32

config/AEPSA/BJTaxi-outflow.yaml

@@ -11,10 +11,10 @@ data:
   column_wise: false
   days_per_week: 7
   default_graph: true
-  horizon: 12
-  lag: 12
+  horizon: 24
+  lag: 24
   normalizer: std
-  num_nodes: 142
+  num_nodes: 1024
   steps_per_day: 48
   test_ratio: 0.2
   tod: false
@@ -24,8 +24,8 @@ data:
   batch_size: 32
 
 model:
-  pred_len: 12
-  seq_len: 12
+  pred_len: 24
+  seq_len: 24
   patch_len: 6 
   stride: 7
   dropout: 0.2
@@ -36,6 +36,7 @@ model:
   n_heads: 1
   input_dim: 1
   word_num: 1000
+  num_nodes: 1024
 
 train:
   batch_size: 32

config/AEPSA/METR-LA.yaml

@@ -36,6 +36,7 @@ model:
   n_heads: 1
   input_dim: 1
   word_num: 1000
+  num_nodes: 207
 
 train:
   batch_size: 16

config/AEPSA/NYCBike-inflow.yaml

@@ -11,10 +11,10 @@ data:
   column_wise: false
   days_per_week: 7
   default_graph: true
-  horizon: 12
-  lag: 12
+  horizon: 24
+  lag: 24
   normalizer: std
-  num_nodes: 200
+  num_nodes: 128
   steps_per_day: 24
   test_ratio: 0.2
   tod: false
@@ -24,8 +24,8 @@ data:
   batch_size: 32
 
 model:
-  pred_len: 12
-  seq_len: 12
+  pred_len: 24
+  seq_len: 24
   patch_len: 6 
   stride: 7
   dropout: 0.2
@@ -36,6 +36,7 @@ model:
   n_heads: 1
   input_dim: 1
   word_num: 1000
+  num_nodes: 128
 
 train:
   batch_size: 32

config/AEPSA/NYCBike-outflow.yaml

@@ -11,10 +11,10 @@ data:
   column_wise: false
   days_per_week: 7
   default_graph: true
-  horizon: 12
-  lag: 12
+  horizon: 24
+  lag: 24
   normalizer: std
-  num_nodes: 200
+  num_nodes: 128
   steps_per_day: 24
   test_ratio: 0.2
   tod: false
@@ -24,8 +24,8 @@ data:
   batch_size: 32
 
 model:
-  pred_len: 12
-  seq_len: 12
+  pred_len: 24
+  seq_len: 24
   patch_len: 6 
   stride: 7
   dropout: 0.2
@@ -36,6 +36,7 @@ model:
   n_heads: 1
   input_dim: 1
   word_num: 1000
+  num_nodes: 128
 
 train:
   batch_size: 32

config/AEPSA/PEMS-BAY.yaml

@@ -36,6 +36,7 @@ model:
   n_heads: 1
   input_dim: 1
   word_num: 1000
+  num_nodes: 325
 
 train:
   batch_size: 16