czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

REPST #3

Merged
czzhangheng merged 42 commits from REPST into main 2025-12-20 16:03:22 +08:00
Conversation 0 Commits 42 Files Changed 465 +10747 -3684
czzhangheng commented 2025-12-20 16:02:59 +08:00
Owner
Copy Link

impl: ASTRA，iTransformer，FPT， PatchTST，MTGNN

update: Trainer, Dataloader

impl: ASTRA，iTransformer，FPT， PatchTST，MTGNN update: Trainer, Dataloader
czzhangheng added 42 commits 2025-12-20 16:02:59 +08:00
3095b7435b refactor: 重构数据加载器和训练器代码，优化代码结构和可读性 
重构数据加载器模块，使用字典映射替代switch-case结构
简化训练器逻辑，合并重复代码，提高可维护性
优化日志时间格式，缩短显示长度
调整训练配置，减少默认epoch数并启用GPU训练
统一数据加载方式，提取公共方法减少重复代码
56b09ea8ac feat: 重构训练器和配置结构，优化模型训练流程 
refactor(trainer): 重构Trainer类结构，拆分初始化方法
perf(trainer): 优化训练循环和评估逻辑
style(config): 统一配置文件命名和结构
fix(trainer): 修复形状检查逻辑和调试模式处理
docs: 更新README和注释说明
b6d4f5daf5 refactor(dataloader): 重构数据加载器代码，优化滑动窗口生成和归一化处理 
重构PeMSD和EXP数据加载器，使用numpy的stride_tricks实现高效滑动窗口
统一数据预处理流程，简化代码结构并提高可维护性
优化归一化处理，支持多scaler通道独立处理
5e52f23c8d fix(config): 修正配置文件命名不一致问题并更新训练参数 
refactor(trainer): 重构训练器代码，优化反归一化处理和形状转换逻辑

style(trainer): 简化代码格式，提高可读性

chore: 更新训练脚本中的模型和数据集列表
659b41f612 refactor(config): 统一配置文件命名并调整模型参数 
更新多个模型的配置文件命名格式，统一使用大驼峰格式
调整SolarEnergy和BJTaxi数据集的输入维度和批量大小
删除旧命名格式的配置文件并添加新的配置文件
修改训练脚本中的模型和数据集列表用于调试
85257bc61c refactor(trainer): 优化训练器代码结构并添加进度条显示 
调整训练器代码结构，减少冗余代码，提高可读性
为训练过程添加tqdm进度条，实时显示loss信息
统一TRY环境变量的输出格式
简化日志记录和模型保存逻辑
czzhangheng merged commit 5a7ec07a12 into main 2025-12-20 16:03:22 +08:00
Sign in to join this conversation.
Reviewers
No reviewers
Labels
Clear labels
No items
No Label
Milestone
Clear milestone
No items
No Milestone
Projects
Clear projects
No project
Assignees
Clear assignees
No Assignees
1 Participants
Notifications
Due Date
The due date is invalid or out of range. Please use the format 'yyyy-mm-dd'.

No due date set.

Dependencies

No dependencies set.

Reference: czzhangheng/TrafficWheel#3
No description provided.
Delete Branch "REPST"

Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?