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_step1: 60,75,90,120
time_dim: 16
train:
loss_func: mae
seed: 10

4
config/PDG2SEQ/PEMSD7.yaml
View File

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

4
config/PDG2SEQ/PEMSD8.yaml
View File

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