Commit Graph

32 Commits

Author SHA1 Message Date
czzhangheng b6d4f5daf5 refactor(dataloader): 重构数据加载器代码,优化滑动窗口生成和归一化处理
重构PeMSD和EXP数据加载器,使用numpy的stride_tricks实现高效滑动窗口
统一数据预处理流程,简化代码结构并提高可维护性
优化归一化处理,支持多scaler通道独立处理
2025-12-15 20:54:20 +08:00
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 44ffe94c95 更新iTransformer, HI配置。更新TS数据集载入方式 2025-12-10 11:01:00 +08:00
czzhangheng 5c2380ae21 适配HI 2025-12-09 17:49:18 +08:00
czzhangheng b57fcef039 新增SolarEnergy-iTransformer配置 2025-12-09 16:56:30 +08:00
czzhangheng faeb90e734 实现iTransformer 2025-12-09 16:12:05 +08:00
czzhangheng 4984d24506 实现iTransformer 2025-12-09 16:11:49 +08:00
czzhangheng 5f9ab76a27 兼容NYCBike 2025-11-23 21:24:10 +08:00
czzhangheng 475a4788cd 兼容BJTaxi 2025-11-23 19:04:50 +08:00
czzhangheng 7055a6da64 更新正确的AirQuality数据集 2025-11-20 22:15:48 +08:00
czzhangheng a46edc79a5 兼容BeijingAirQuality。重构data,需要更新pip requirement 2025-11-20 20:19:17 +08:00
czzhangheng 9911caa3d8 兼容solarEnergy 2025-11-20 11:28:58 +08:00
czzhangheng a9e7cd5d3b update data_selector 2025-11-20 10:51:32 +08:00
czzhangheng 76acc89499 兼容METR-LA 2025-11-20 10:48:05 +08:00
czzhangheng 8b7e13df30 优化trainer和run 2025-11-18 10:01:01 +08:00
czzhangheng 9b3bb44552 修复配置, trainer 2025-11-09 22:30:23 +08:00
czzhangheng 241274e60f 删除冗余代码。重构代码 2025-11-09 20:40:19 +08:00
czzhangheng 2685d049d7 兼容pems-bay 2025-11-09 18:51:47 +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 1be0b59344 新增EXP数据加载器、模型和训练器,支持周期性数据处理和动态图构建 2025-05-08 22:43:26 +08:00
czzhangheng eb8684bf91 收到垃圾堆 2025-04-22 14:54:39 +08:00
czzhangheng 86fabd4ca7 增加 exp9 混合专家
exp8 动态图manba
2025-04-17 18:41:57 +08:00
czzhangheng 0a9ac1a025 fix bugs 2025-03-03 16:31:04 +08:00
HengZhang 44517219b7 add dl function 2025-03-03 11:34:36 +08:00
HengZhang be5e810c54 mv dir name 2025-03-03 10:29:42 +08:00
HengZhang 26758e761b init 2025-03-02 23:41:12 +08:00