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

REPST #3

Merged
czzhangheng merged 42 commits from REPST into main 2025-12-20 16:03:22 +08:00
Conversation 0 Commits 42 Files Changed 465 +3 -3
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit b97111f5ea - Show all commits

2
config/REPST/AirQuality.yaml
View File

@ -50,7 +50,7 @@ train:
mae_thresh: None
mape_thresh: 0.001
max_grad_norm: 5
output_dim: 3
output_dim: 6
plot: false
real_value: true
weight_decay: 0

4
train.py
View File

@ -89,8 +89,8 @@ def main(model, data, debug=False):
if __name__ == "__main__":
# 调试用
model_list = ["iTransformer", "PatchTST", "HI"]
# model_list = ["ASTRA_v2", "GWN", "REPST", "STAEFormer", "MTGNN"]
# model_list = ["iTransformer", "PatchTST", "HI"]
model_list = ["ASTRA_v3", "ASTRA_v2", "ASTRA", "REPST", "STAEFormer", "MTGNN", "iTransformer", "PatchTST", "HI"]
# model_list = ["MTGNN"]
# dataset_list = ["AirQuality", "SolarEnergy", "PEMS-BAY", "METR-LA", "BJTaxi-InFlow", "BJTaxi-OutFlow", "NYCBike-InFlow", "NYCBike-OutFlow"]
# dataset_list = ["AirQuality"]