Commit Graph

96 Commits

Author SHA1 Message Date
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
czzhangheng 2ba061e57a 修复confuse_layer硬编码bug 2025-11-11 17:26:05 +08:00
czzhangheng d0af46ea5f 修复缺库问题 2025-11-10 11:13:39 +08:00
czzhangheng f3480fccdc STID配置更新 2025-11-10 09:34:52 +08:00
czzhangheng 7da402d5e0 Merge remote-tracking branch 'origin/main' into REPST
更新main支的修复
2025-11-10 09:34:04 +08:00
czzhangheng 9b3bb44552 修复配置, trainer 2025-11-09 22:30:23 +08:00
czzhangheng 2800f66dfe 删除冗发余代码。重构代码 2025-11-09 20:43:00 +08:00
czzhangheng 241274e60f 删除冗余代码。重构代码 2025-11-09 20:40:19 +08:00
czzhangheng 2685d049d7 兼容pems-bay 2025-11-09 18:51:47 +08:00
czzhangheng 162bc8dc53 REPST_README 2025-11-09 16:45:45 +08:00
czzhangheng c6e5e904ea 修复未使用的导入 2025-11-09 16:39:59 +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 b02c9c91d7 删除多余空行 2025-11-08 21:26:56 +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 b26e5a823c 修复下载数据集失败的问题,更新依赖 2025-11-08 19:24:54 +08:00
czzhangheng 29fd709c8c 新增多个PEMS数据集配置文件,包含PEMSD3、PEMSD4、PEMSD7、PEMSD8及STAWnet、STGNRDE、ST_SSL模型的相关配置,优化模型训练参数设置。 2025-08-19 00:07:37 +08:00