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 |