 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 |
|
czzhangheng
|
b820b867fb
|
新模型
|
2025-08-19 00:07:13 +08:00 |
|
czzhangheng
|
04f65856f6
|
更新 .gitignore
|
2025-08-19 00:01:00 +08:00 |
|
czzhangheng
|
42b9cd3342
|
Merge remote-tracking branch 'origin/main'
|
2025-08-18 21:07:49 +08:00 |
|
czzhangheng
|
af795043c8
|
Merge branch 'main' into dev
# Conflicts:
# run.py
# trainer/Trainer.py
|
2025-08-18 21:07:16 +08:00 |
harry.zhang
|
8025a46baa
|
更新.gitignore以忽略Result.xlsx文件，修改run.py以优先使用macOS的MPS设备，优化设备设置逻辑
|
2025-08-18 14:32:20 +08:00 |
|
czzhangheng
|
bacdce7c72
|
trainer添加效率统计
|
2025-08-17 14:03:26 +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
|
c15cf605be
|
添加md结果
|
2025-04-26 15:37:40 +08:00 |
|
czzhangheng
|
5f8c31af2e
|
添加STIDGCN
|
2025-04-23 23:24:43 +08:00 |
|
czzhangheng
|
e826240a5e
|
Merge remote-tracking branch 'origin/main'
# Conflicts:
# model/model_selector.py
|
2025-04-23 23:24:05 +08:00 |