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

更新配置，METR-LA -STAEFormer

This commit is contained in:
czzhangheng 2025-12-03 18:51:54 +08:00
parent 9b1cf5f0ce
commit 07d7d43857
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
config/STAEFormer/METR-LA.yaml
View File

@ -9,9 +9,9 @@ data:
batch_size: 16 batch_size: 16
column_wise: false column_wise: false
days_per_week: 7 days_per_week: 7
horizon: 12 horizon: 24
input_dim: 1 input_dim: 1
lag: 12 lag: 24
normalizer: std normalizer: std
num_nodes: 207 num_nodes: 207
steps_per_day: 288 steps_per_day: 288
@ -23,13 +23,13 @@ model:
dow_embedding_dim: 24 dow_embedding_dim: 24
dropout: 0.1 dropout: 0.1
feed_forward_dim: 256 feed_forward_dim: 256
in_steps: 12 in_steps: 24
input_dim: 1 input_dim: 1
input_embedding_dim: 24 input_embedding_dim: 24
num_heads: 4 num_heads: 4
num_layers: 3 num_layers: 3
num_nodes: 207 num_nodes: 207
out_steps: 12 out_steps: 24
output_dim: 1 output_dim: 1
spatial_embedding_dim: 0 spatial_embedding_dim: 0
steps_per_day: 288 steps_per_day: 288