 czzhangheng
|
85257bc61c
|
refactor(trainer): 优化训练器代码结构并添加进度条显示
调整训练器代码结构，减少冗余代码，提高可读性
为训练过程添加tqdm进度条，实时显示loss信息
统一TRY环境变量的输出格式
简化日志记录和模型保存逻辑
|
2025-12-16 16:40:40 +08:00 |
|
czzhangheng
|
659b41f612
|
refactor(config): 统一配置文件命名并调整模型参数
更新多个模型的配置文件命名格式，统一使用大驼峰格式
调整SolarEnergy和BJTaxi数据集的输入维度和批量大小
删除旧命名格式的配置文件并添加新的配置文件
修改训练脚本中的模型和数据集列表用于调试
|
2025-12-15 21:33:28 +08:00 |
|
czzhangheng
|
5e52f23c8d
|
fix(config): 修正配置文件命名不一致问题并更新训练参数
refactor(trainer): 重构训练器代码，优化反归一化处理和形状转换逻辑
style(trainer): 简化代码格式，提高可读性
chore: 更新训练脚本中的模型和数据集列表
|
2025-12-15 21:23:04 +08:00 |
|
czzhangheng
|
56b09ea8ac
|
feat: 重构训练器和配置结构，优化模型训练流程
refactor(trainer): 重构Trainer类结构，拆分初始化方法
perf(trainer): 优化训练循环和评估逻辑
style(config): 统一配置文件命名和结构
fix(trainer): 修复形状检查逻辑和调试模式处理
docs: 更新README和注释说明
|
2025-12-15 19:55:57 +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
|
600420e8df
|
impl mtgnn
|
2025-12-10 23:31:17 +08:00 |
|
czzhangheng
|
4ccb029d7e
|
impl PatchTST
|
2025-12-10 21:53:46 +08:00 |
|
czzhangheng
|
d8f4cc5825
|
简化trainer，修复设备bug，新增批量运行脚本
|
2025-12-10 21:08:20 +08:00 |
|
czzhangheng
|
44ffe94c95
|
更新iTransformer, HI配置。更新TS数据集载入方式
|
2025-12-10 11:01:00 +08:00 |
|
czzhangheng
|
560d24e5a8
|
更新v2配置
|
2025-12-10 10:39:41 +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
|
e4a7884c98
|
更名ASTRA配置
|
2025-12-09 14:08:59 +08:00 |
|
czzhangheng
|
9c76975056
|
更名ASTRA
|
2025-12-09 14:07:38 +08:00 |
|
czzhangheng
|
1598918112
|
更改v3_PEWSBAY配置
|
2025-12-06 19:49:31 +08:00 |
|
czzhangheng
|
865c5a3082
|
ASTRA-V3
|
2025-12-06 19:47:33 +08:00 |
|
czzhangheng
|
07d7d43857
|
更新配置，METR-LA -STAEFormer
|
2025-12-03 18:51:54 +08:00 |
|
czzhangheng
|
440cb6936b
|
兼容STAEFormer
|
2025-12-03 17:12:46 +08:00 |
|
czzhangheng
|
a9313390ac
|
适配GraphWaveNet
|
2025-12-03 12:05:02 +08:00 |
|
czzhangheng
|
140ead3975
|
为model添加num_nodes
|
2025-12-03 08:40:00 +08:00 |
|
czzhangheng
|
e50f443470
|
add astra-pemsbay v2
|
2025-12-02 09:40:24 +08:00 |
|
czzhangheng
|
c4414dd5d9
|
fix twdgcn config
|
2025-12-02 09:12:09 +08:00 |
|
czzhangheng
|
f64144f5c1
|
为其他模型添加配置文件
|
2025-12-01 20:45:30 +08:00 |
|
czzhangheng
|
22e19f8804
|
更新新数据集在其他模型下的配置
|
2025-12-01 20:12:48 +08:00 |
|
czzhangheng
|
6e94ae90d2
|
修复冗余参数项
|
2025-12-01 19:52:08 +08:00 |
|
czzhangheng
|
a1a50880dc
|
删除冗余参数项
|
2025-12-01 19:13:23 +08:00 |
|
czzhangheng
|
d7d428e434
|
更改配置为字典序
|
2025-12-01 19:04:14 +08:00 |
|
czzhangheng
|
f3b5bdfc28
|
更新v2可读性
|
2025-11-28 21:32:49 +08:00 |
|
czzhangheng
|
0ad1494aec
|
修复模型 配置错误
|
2025-11-27 19:21:01 +08:00 |
|
czzhangheng
|
35310a4f4a
|
修复模型、配置错误
|
2025-11-27 19:20:59 +08:00 |
|
czzhangheng
|
2f58fc9348
|
aepsa_v2 solarenergy
|
2025-11-27 11:40:09 +08:00 |
|
czzhangheng
|
d5e9e52ee1
|
AEPSA_V2
|
2025-11-27 09:32:37 +08:00 |
|
czzhangheng
|
c22d946393
|
mask_mae
|
2025-11-27 08:36:07 +08:00 |
|
czzhangheng
|
4019645f5d
|
add STID-METR-LA
|
2025-11-25 20:03:19 +08:00 |
|
czzhangheng
|
bcedc0c62a
|
更新配置
|
2025-11-25 11:25:30 +08:00 |
|
czzhangheng
|
9c50c30918
|
AEPSA v0.1
|
2025-11-24 21:50:24 +08:00 |
|
czzhangheng
|
88daa53dd5
|
更新pip, STID配置
|
2025-11-24 09:42:33 +08:00 |
|
czzhangheng
|
59fb75cd3c
|
修正配置
|
2025-11-23 21:26:45 +08:00 |
|
czzhangheng
|
5f9ab76a27
|
兼容NYCBike
|
2025-11-23 21:24:10 +08:00 |
|
czzhangheng
|
5cd81f4d4c
|
Merge branch 'REPST' of https://github.zhang-heng.com/czzhangheng/TrafficWheel into REPST
|
2025-11-23 19:10:51 +08:00 |
|
czzhangheng
|
95f81425b0
|
兼容consineAnnealineLR
|
2025-11-23 19:06:48 +08:00 |
|
czzhangheng
|
475a4788cd
|
兼容BJTaxi
|
2025-11-23 19:04:50 +08:00 |
|
czzhangheng
|
d06cf4e0aa
|
修复配置错误
|
2025-11-21 09:32:52 +08:00 |
|
czzhangheng
|
7055a6da64
|
更新正确的AirQuality数据集
|
2025-11-20 22:15:48 +08:00 |
|
czzhangheng
|
96f2ea1239
|
修复反归一化错误
|
2025-11-20 20:50:35 +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 |