Commit Graph

63 Commits

Author SHA1 Message Date
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
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 38431fb4c1 校准参数 2025-11-12 11:01:21 +08:00
czzhangheng a3fb2d5ec6 更新AEPSA框架,待修改文件 2025-11-11 21:29:52 +08:00
czzhangheng 2ba061e57a 修复confuse_layer硬编码bug 2025-11-11 17:26:05 +08:00
czzhangheng f3480fccdc STID配置更新 2025-11-10 09:34:52 +08:00
czzhangheng 9b3bb44552 修复配置, trainer 2025-11-09 22:30:23 +08:00
czzhangheng 2685d049d7 兼容pems-bay 2025-11-09 18:51:47 +08:00
czzhangheng cf47acf2bd 修复test bug 2025-11-09 16:38:04 +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 4f7fb52707 更新EXP模型,添加周期性数据处理和时间特征,优化数据加载器和训练器,支持新的EXP32模型结构 2025-05-08 22:43:33 +08:00
czzhangheng e826240a5e Merge remote-tracking branch 'origin/main'
# Conflicts:
#	model/model_selector.py
2025-04-23 23:24:05 +08:00
czzhangheng 97eb39073a 添加STIDGCN 2025-04-23 23:22:50 +08:00