TrafficWheel/config
czzhangheng 5e52f23c8d fix(config): 修正配置文件命名不一致问题并更新训练参数
refactor(trainer): 重构训练器代码,优化反归一化处理和形状转换逻辑

style(trainer): 简化代码格式,提高可读性

chore: 更新训练脚本中的模型和数据集列表
2025-12-15 21:23:04 +08:00
..
AGCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ARIMA 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ASTRA feat: 重构训练器和配置结构,优化模型训练流程 2025-12-15 19:55:57 +08:00
ASTRA_v2 feat: 重构训练器和配置结构,优化模型训练流程 2025-12-15 19:55:57 +08:00
ASTRA_v3 feat: 重构训练器和配置结构,优化模型训练流程 2025-12-15 19:55:57 +08:00
DCRNN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
DDGCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
DSANET 为model添加num_nodes 2025-12-03 08:40:00 +08:00
EXP 为model添加num_nodes 2025-12-03 08:40:00 +08:00
EXPB 为model添加num_nodes 2025-12-03 08:40:00 +08:00
GWN 适配GraphWaveNet 2025-12-03 12:05:02 +08:00
HI fix(config): 修正配置文件命名不一致问题并更新训练参数 2025-12-15 21:23:04 +08:00
Informer fix(config): 修正配置文件命名不一致问题并更新训练参数 2025-12-15 21:23:04 +08:00
MTGNN feat: 重构训练器和配置结构,优化模型训练流程 2025-12-15 19:55:57 +08:00
MegaCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
NLT 为model添加num_nodes 2025-12-03 08:40:00 +08:00
PDG2SEQ 为model添加num_nodes 2025-12-03 08:40:00 +08:00
PatchTST 改进trainer,使用统一的loader而不是TSLoader,仅在Trainer上做了shape变换,确定尽可能少改动数据。 2025-12-14 17:47:38 +08:00
REPST feat: 重构训练器和配置结构,优化模型训练流程 2025-12-15 19:55:57 +08:00
STAEFormer 更新配置,METR-LA -STAEFormer 2025-12-03 18:51:54 +08:00
STAWnet 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STFGNN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGNCDE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGNRDE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGODE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STID 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STIDGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STMLP 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STSGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ST_SSL 为model添加num_nodes 2025-12-03 08:40:00 +08:00
TCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
TWDGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
iTransformer fix(config): 修正配置文件命名不一致问题并更新训练参数 2025-12-15 21:23:04 +08:00
args_parser.py 简化trainer,修复设备bug,新增批量运行脚本 2025-12-10 21:08:20 +08:00