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

fix PDG2SEQ config

This commit is contained in:
czzhangheng 2025-03-10 19:05:26 +08:00
parent ae7e91d162
commit dcd4f99f9c
3 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
config/PDG2SEQ/PEMSD4.yaml
View File

@ -25,6 +25,7 @@ model:
lr_decay_step: 1500 lr_decay_step: 1500
lr_decay_step1: 60,75,90,120 lr_decay_step1: 60,75,90,120
time_dim: 16 time_dim: 16
train: train:
loss_func: mae loss_func: mae
seed: 10 seed: 10

4
config/PDG2SEQ/PEMSD7.yaml
View File

@ -12,9 +12,11 @@ data:
test_ratio: 0.2 test_ratio: 0.2
tod: false tod: false
val_ratio: 0.2 val_ratio: 0.2
log: log:
log_step: 3000 log_step: 3000
plot: false plot: false
model: model:
cheb_k: 2 cheb_k: 2
embed_dim: 12 embed_dim: 12
@ -27,9 +29,11 @@ model:
lr_decay_step: 12000 lr_decay_step: 12000
lr_decay_step1: 80,100,120 lr_decay_step1: 80,100,120
time_dim: 20 time_dim: 20
test: test:
mae_thresh: None mae_thresh: None
mape_thresh: 0.0 mape_thresh: 0.0
train: train:
batch_size: 16 batch_size: 16
early_stop: true early_stop: true

4
config/PDG2SEQ/PEMSD8.yaml
View File

@ -12,9 +12,11 @@ data:
test_ratio: 0.2 test_ratio: 0.2
tod: false tod: false
val_ratio: 0.2 val_ratio: 0.2
log: log:
log_step: 2000 log_step: 2000
plot: false plot: false
model: model:
cheb_k: 2 cheb_k: 2
embed_dim: 12 embed_dim: 12
@ -27,9 +29,11 @@ model:
lr_decay_step: 2000 lr_decay_step: 2000
lr_decay_step1: 50,75 lr_decay_step1: 50,75
time_dim: 16 time_dim: 16
test: test:
mae_thresh: None mae_thresh: None
mape_thresh: 0.001 mape_thresh: 0.001
train: train:
batch_size: 64 batch_size: 64
early_stop: true early_stop: true