 czzhangheng
|
9d3293cef7
|
impl STNorm
|
2025-12-20 15:45:13 +08:00 |
|
czzhangheng
|
cc63e5078f
|
impl FPT
|
2025-12-19 10:19:17 +08:00 |
|
czzhangheng
|
21bc05e763
|
修复TSLoader Bug
|
2025-12-17 17:13:16 +08:00 |
|
czzhangheng
|
1a13a32688
|
修复配置。ASTRA v2 or v3使用硬参数，确保配置文件完整
|
2025-12-16 21:47:17 +08:00 |
|
czzhangheng
|
b38e4a5da2
|
修复astra bug
|
2025-12-16 17:31:57 +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
|
5827554c73
|
改进模型注册，动态注册
|
2025-12-14 17:48:37 +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
|
560d24e5a8
|
更新v2配置
|
2025-12-10 10:39:41 +08:00 |
|
czzhangheng
|
5c2380ae21
|
适配HI
|
2025-12-09 17:49:18 +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
|
9c76975056
|
更名ASTRA
|
2025-12-09 14:07:38 +08:00 |
|
czzhangheng
|
aed1e53f0f
|
更名ASTRA
|
2025-12-09 14:06:56 +08:00 |
|
czzhangheng
|
f899f50b16
|
变量名更改
|
2025-12-06 19:48:27 +08:00 |
|
czzhangheng
|
865c5a3082
|
ASTRA-V3
|
2025-12-06 19:47:33 +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
|
c4414dd5d9
|
fix twdgcn config
|
2025-12-02 09:12:09 +08:00 |
|
czzhangheng
|
f3b5bdfc28
|
更新v2可读性
|
2025-11-28 21:32:49 +08:00 |
|
czzhangheng
|
35310a4f4a
|
修复模型、配置错误
|
2025-11-27 19:20:59 +08:00 |
|
czzhangheng
|
d5e9e52ee1
|
AEPSA_V2
|
2025-11-27 09:32:37 +08:00 |
|
czzhangheng
|
7c984c6fd2
|
AEPSA v0.1
|
2025-11-24 21:53:01 +08:00 |
|
czzhangheng
|
9c50c30918
|
AEPSA v0.1
|
2025-11-24 21:50:24 +08:00 |
|
czzhangheng
|
5f9ab76a27
|
兼容NYCBike
|
2025-11-23 21:24:10 +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
|
8b7e13df30
|
优化trainer和run
|
2025-11-18 10:01:01 +08:00 |
|
czzhangheng
|
e61706c391
|
可调word_num
|
2025-11-12 17:12:30 +08:00 |
|
czzhangheng
|
15f083c3d9
|
修复permute的bug
|
2025-11-12 16:40:13 +08:00 |
|
czzhangheng
|
095e8c60dc
|
修复没permute的bug
|
2025-11-12 16:40:09 +08:00 |
|
czzhangheng
|
a3fb2d5ec6
|
更新AEPSA框架，待修改文件
|
2025-11-11 21:29:52 +08:00 |
|
czzhangheng
|
6657743afe
|
清理repst冗余代码
|
2025-11-11 21:00:36 +08:00 |
|
czzhangheng
|
2ba061e57a
|
修复confuse_layer硬编码bug
|
2025-11-11 17:26:05 +08:00 |
|
czzhangheng
|
d0af46ea5f
|
修复缺库问题
|
2025-11-10 11:13:39 +08:00 |
|
czzhangheng
|
9b3bb44552
|
修复配置, trainer
|
2025-11-09 22:30:23 +08:00 |
|
czzhangheng
|
241274e60f
|
删除冗余代码。重构代码
|
2025-11-09 20:40:19 +08:00 |
|
czzhangheng
|
a3e43fc6df
|
添加支持REPST In D8
|
2025-11-09 16:27:32 +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
|
b820b867fb
|
新模型
|
2025-08-19 00:07:13 +08:00 |
|
czzhangheng
|
bef30b9c2f
|
解决合并冲突，整合dev和main分支的更改
|
2025-05-14 13:13:11 +08:00 |
|
czzhangheng
|
61565cd33a
|
为GraphWaveNet模型添加详细注释，增强代码可读性和可维护性
|
2025-05-14 13:09:20 +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
|
5f8c31af2e
|
添加STIDGCN
|
2025-04-23 23:24:43 +08:00 |