diff --git a/config/AEPSA/AirQuality.yaml b/config/AEPSA/AirQuality.yaml index 652a544..c48bbb5 100644 --- a/config/AEPSA/AirQuality.yaml +++ b/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 diff --git a/config/AEPSA/BJTaxi-Inflow.yaml b/config/AEPSA/BJTaxi-Inflow.yaml index 771119b..2384cf5 100644 --- a/config/AEPSA/BJTaxi-Inflow.yaml +++ b/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 diff --git a/config/AEPSA/BJTaxi-outflow.yaml b/config/AEPSA/BJTaxi-outflow.yaml index 936b1a3..2ce962b 100644 --- a/config/AEPSA/BJTaxi-outflow.yaml +++ b/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 diff --git a/config/AEPSA/METR-LA.yaml b/config/AEPSA/METR-LA.yaml index b24f448..2b1948b 100644 --- a/config/AEPSA/METR-LA.yaml +++ b/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 diff --git a/config/AEPSA/NYCBike-inflow.yaml b/config/AEPSA/NYCBike-inflow.yaml index ee0bd06..fd5ed41 100644 --- a/config/AEPSA/NYCBike-inflow.yaml +++ b/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 diff --git a/config/AEPSA/NYCBike-outflow.yaml b/config/AEPSA/NYCBike-outflow.yaml index a620e0e..ba7b014 100644 --- a/config/AEPSA/NYCBike-outflow.yaml +++ b/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 diff --git a/config/AEPSA/PEMS-BAY.yaml b/config/AEPSA/PEMS-BAY.yaml index 99d58a1..84c5902 100755 --- a/config/AEPSA/PEMS-BAY.yaml +++ b/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