Commit Graph

12 Commits

Author SHA1 Message Date
czzhangheng 3095b7435b refactor: 重构数据加载器和训练器代码,优化代码结构和可读性
重构数据加载器模块,使用字典映射替代switch-case结构
简化训练器逻辑,合并重复代码,提高可维护性
优化日志时间格式,缩短显示长度
调整训练配置,减少默认epoch数并启用GPU训练
统一数据加载方式,提取公共方法减少重复代码
2025-12-15 01:38:47 +08:00
czzhangheng 9147803c2b 改进trainer,使用统一的loader而不是TSLoader,仅在Trainer上做了shape变换,确定尽可能少改动数据。 2025-12-14 17:47:38 +08:00
czzhangheng 19fd7622a3 兼容InFormer 2025-12-11 23:16:25 +08:00
czzhangheng 4ccb029d7e impl PatchTST 2025-12-10 21:53:46 +08:00
czzhangheng 5c2380ae21 适配HI 2025-12-09 17:49:18 +08:00
czzhangheng 4984d24506 实现iTransformer 2025-12-09 16:11:49 +08:00
czzhangheng 221b67aa53 标准化PEP8格式 2025-11-08 21:26:03 +08:00
czzhangheng 9f9abd9d1b 预备升级为2.0 2025-11-08 21:05:52 +08:00
czzhangheng 29fd709c8c 新增多个PEMS数据集配置文件,包含PEMSD3、PEMSD4、PEMSD7、PEMSD8及STAWnet、STGNRDE、ST_SSL模型的相关配置,优化模型训练参数设置。 2025-08-19 00:07:37 +08:00
czzhangheng 4f7fb52707 更新EXP模型,添加周期性数据处理和时间特征,优化数据加载器和训练器,支持新的EXP32模型结构 2025-05-08 22:43:33 +08:00
czzhangheng eb8684bf91 收到垃圾堆 2025-04-22 14:54:39 +08:00
HengZhang 26758e761b init 2025-03-02 23:41:12 +08:00