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

new-model,MEGA stssl

This commit is contained in:
czzhangheng 2025-08-18 21:57:45 +08:00
parent 538548db0b
commit 6545482a58
9 changed files with 424 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
config/MegaCRN/PEMSD3.yaml Normal file
View File

@ -0,0 +1,52 @@
data:
num_nodes: 358
lag: 12
horizon: 12
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
model:
input_dim: 1
output_dim: 1
rnn_units: 64
num_layers: 1
cheb_k: 3
ycov_dim: 1
mem_num: 20
mem_dim: 64
cl_decay_steps: 2000
use_curriculum_learning: True
train:
loss_func: mae
seed: 10
batch_size: 64
epochs: 200
lr_init: 0.003
weight_decay: 0
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "5,20,40,70"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
test:
mae_thresh: null
mape_thresh: 0.0
log:
log_step: 200
plot: False

52
config/MegaCRN/PEMSD4.yaml Normal file
View File

@ -0,0 +1,52 @@
data:
num_nodes: 307
lag: 12
horizon: 12
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
model:
input_dim: 1
output_dim: 1
rnn_units: 64
num_layers: 1
cheb_k: 3
ycov_dim: 1
mem_num: 20
mem_dim: 64
cl_decay_steps: 2000
use_curriculum_learning: True
train:
loss_func: mae
seed: 10
batch_size: 64
epochs: 200
lr_init: 0.003
weight_decay: 0
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "5,20,40,70"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
test:
mae_thresh: null
mape_thresh: 0.0
log:
log_step: 200
plot: False

52
config/MegaCRN/PEMSD7.yaml Normal file
View File

@ -0,0 +1,52 @@
data:
num_nodes: 883
lag: 12
horizon: 12
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
model:
input_dim: 1
output_dim: 1
rnn_units: 64
num_layers: 1
cheb_k: 3
ycov_dim: 1
mem_num: 20
mem_dim: 64
cl_decay_steps: 2000
use_curriculum_learning: True
train:
loss_func: mae
seed: 10
batch_size: 64
epochs: 300
lr_init: 0.003
weight_decay: 0
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "5,20,40,70"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
test:
mae_thresh: null
mape_thresh: 0.0
log:
log_step: 200
plot: False

52
config/MegaCRN/PEMSD8.yaml Normal file
View File

@ -0,0 +1,52 @@
data:
num_nodes: 170
lag: 12
horizon: 12
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
model:
input_dim: 1
output_dim: 1
rnn_units: 64
num_layers: 1
cheb_k: 3
ycov_dim: 1
mem_num: 20
mem_dim: 64
cl_decay_steps: 2000
use_curriculum_learning: True
train:
loss_func: mae
seed: 10
batch_size: 64
epochs: 300
lr_init: 0.003
weight_decay: 0
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "5,20,40,70"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
test:
mae_thresh: null
mape_thresh: 0.0
log:
log_step: 200
plot: False

53
config/ST_SSL/PEMSD3.yaml Normal file
View File

@ -0,0 +1,53 @@
data:
num_nodes: 358
lag: 12
horizon: 12
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
model:
input_dim: 1
output_dim: 1
n_his: 12
d_model: 64
dropout: 0.1
nmb_prototype: 10
shm_temp: 0.1
yita: 0.5
percent: 0.1
gso_type: sym_norm_lap
graph_conv_type: cheb_graph_conv
train:
loss_func: mae
seed: 10
batch_size: 64
epochs: 200
lr_init: 0.003
weight_decay: 0
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "5,20,40,70"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
test:
mae_thresh: null
mape_thresh: 0.0
log:
log_step: 200
plot: False

53
config/ST_SSL/PEMSD4.yaml Normal file
View File

@ -0,0 +1,53 @@
data:
num_nodes: 307
lag: 12
horizon: 12
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
model:
input_dim: 1
output_dim: 1
n_his: 12
d_model: 64
dropout: 0.1
nmb_prototype: 10
shm_temp: 0.1
yita: 0.5
percent: 0.1
gso_type: sym_norm_lap
graph_conv_type: cheb_graph_conv
train:
loss_func: mae
seed: 10
batch_size: 64
epochs: 300
lr_init: 0.003
weight_decay: 0
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "5,20,40,70"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
test:
mae_thresh: null
mape_thresh: 0.0
log:
log_step: 200
plot: False

53
config/ST_SSL/PEMSD7.yaml Normal file
View File

@ -0,0 +1,53 @@
data:
num_nodes: 883
lag: 12
horizon: 12
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
model:
input_dim: 1
output_dim: 1
n_his: 12
d_model: 64
dropout: 0.1
nmb_prototype: 10
shm_temp: 0.1
yita: 0.5
percent: 0.1
gso_type: sym_norm_lap
graph_conv_type: cheb_graph_conv
train:
loss_func: mae
seed: 10
batch_size: 64
epochs: 300
lr_init: 0.003
weight_decay: 0
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "5,20,40,70"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
test:
mae_thresh: null
mape_thresh: 0.0
log:
log_step: 200
plot: False

53
config/ST_SSL/PEMSD8.yaml Normal file
View File

@ -0,0 +1,53 @@
data:
num_nodes: 170
lag: 12
horizon: 12
val_ratio: 0.2
test_ratio: 0.2
tod: False
normalizer: std
column_wise: False
default_graph: True
add_time_in_day: True
add_day_in_week: True
steps_per_day: 288
days_per_week: 7
model:
input_dim: 1
output_dim: 1
n_his: 12
d_model: 64
dropout: 0.1
nmb_prototype: 10
shm_temp: 0.1
yita: 0.5
percent: 0.1
gso_type: sym_norm_lap
graph_conv_type: cheb_graph_conv
train:
loss_func: mae
seed: 10
batch_size: 64
epochs: 300
lr_init: 0.003
weight_decay: 0
lr_decay: False
lr_decay_rate: 0.3
lr_decay_step: "5,20,40,70"
early_stop: True
early_stop_patience: 15
grad_norm: False
max_grad_norm: 5
real_value: True
test:
mae_thresh: null
mape_thresh: 0.0
log:
log_step: 200
plot: False

4
model/model_selector.py
View File

@ -18,6 +18,8 @@ from model.STIDGCN.STIDGCN import STIDGCN
from model.STID.STID import STID
from model.STAEFormer.STAEFormer import STAEformer
from model.EXP.EXP32 import EXP as EXP
from model.MegaCRN.MegaCRNModel import MegaCRNModel
from model.ST_SSL.ST_SSL import STSSLModel
def model_selector(model):
match model['type']:
@ -41,4 +43,6 @@ def model_selector(model):
case 'STID': return STID(model)
case 'STAEFormer': return STAEformer(model)
case 'EXP': return EXP(model)
case 'MegaCRN': return MegaCRNModel(model)
case 'ST_SSL': return STSSLModel(model)