TrafficWheel/config
czzhangheng 56b09ea8ac feat: 重构训练器和配置结构,优化模型训练流程
refactor(trainer): 重构Trainer类结构,拆分初始化方法
perf(trainer): 优化训练循环和评估逻辑
style(config): 统一配置文件命名和结构
fix(trainer): 修复形状检查逻辑和调试模式处理
docs: 更新README和注释说明
2025-12-15 19:55:57 +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 更新iTransformer, HI配置。更新TS数据集载入方式 2025-12-10 11:01:00 +08:00
Informer 兼容InFormer 2025-12-11 23:16:25 +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 改进trainer,使用统一的loader而不是TSLoader,仅在Trainer上做了shape变换,确定尽可能少改动数据。 2025-12-14 17:47:38 +08:00
args_parser.py 简化trainer,修复设备bug,新增批量运行脚本 2025-12-10 21:08:20 +08:00