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.STID.STID import STID
from model.STAEFormer.STAEFormer import STAEformer from model.STAEFormer.STAEFormer import STAEformer
from model.EXP.EXP32 import EXP as EXP 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): def model_selector(model):
match model['type']: match model['type']:
@ -41,4 +43,6 @@ def model_selector(model):
case 'STID': return STID(model) case 'STID': return STID(model)
case 'STAEFormer': return STAEformer(model) case 'STAEFormer': return STAEformer(model)
case 'EXP': return EXP(model) case 'EXP': return EXP(model)
case 'MegaCRN': return MegaCRNModel(model)
case 'ST_SSL': return STSSLModel(model)