czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

更新配置

This commit is contained in:
czzhangheng 2025-11-25 11:25:30 +08:00
parent 7c984c6fd2
commit bcedc0c62a
7 changed files with 27 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
config/AEPSA/AirQuality.yaml
View File

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

11
config/AEPSA/BJTaxi-Inflow.yaml
View File

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

11
config/AEPSA/BJTaxi-outflow.yaml
View File

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

1
config/AEPSA/METR-LA.yaml
View File

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

11
config/AEPSA/NYCBike-inflow.yaml
View File

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

11
config/AEPSA/NYCBike-outflow.yaml
View File

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

1
config/AEPSA/PEMS-BAY.yaml
View File

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