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 |