 czzhangheng
|
9b1cf5f0ce
|
更新测试脚本
|
2025-12-03 18:11:06 +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
|
77a3210475
|
测试所有配置是否正常运行
|
2025-12-01 22:29:52 +08:00 |
|
czzhangheng
|
d4ee8e309e
|
trainer修改
|
2025-12-01 21:36:37 +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
|
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
|
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
|
cf1aa8022f
|
修复beijing taxi检测错误
|
2025-11-23 19:25:51 +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
|
b7ea73bc92
|
更新数据集的检测与下载逻辑
|
2025-11-23 17:58:44 +08:00 |
|
czzhangheng
|
d06cf4e0aa
|
修复配置错误
|
2025-11-21 09:32:52 +08:00 |
|
czzhangheng
|
7055a6da64
|
更新正确的AirQuality数据集
|
2025-11-20 22:15:48 +08:00 |
|
czzhangheng
|
4d87087147
|
测试时汇总样本 使用detach到gpu 避免显存爆炸
|
2025-11-20 21:21:39 +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 |
|
czzhangheng
|
a9e7cd5d3b
|
update data_selector
|
2025-11-20 10:51:32 +08:00 |
|
czzhangheng
|
3b6dc18742
|
Merge branch 'REPST' of https://github.zhang-heng.com/czzhangheng/TrafficWheel into REPST
|
2025-11-20 10:49:49 +08:00 |
|
czzhangheng
|
76acc89499
|
兼容METR-LA
|
2025-11-20 10:48:05 +08:00 |
|
czzhangheng
|
fe3fc186be
|
显示反归一化loss
|
2025-11-18 10:46:08 +08:00 |
|
czzhangheng
|
8b7e13df30
|
优化trainer和run
|
2025-11-18 10:01:01 +08:00 |
|
czzhangheng
|
dceae4b1a3
|
改word_num为1000
|
2025-11-12 17:21:56 +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
|
38431fb4c1
|
校准参数
|
2025-11-12 11:01:21 +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
|
8e53d25ab1
|
将检测数据集逻辑挪到顶部
|
2025-11-11 18:59:12 +08:00 |