TrafficWheel/config
czzhangheng 9147803c2b 改进trainer,使用统一的loader而不是TSLoader,仅在Trainer上做了shape变换,确定尽可能少改动数据。 2025-12-14 17:47:38 +08:00
..
AGCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ARIMA 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ASTRA 更新v2配置 2025-12-10 10:39:41 +08:00
DCRNN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
DDGCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
DSANET 为model添加num_nodes 2025-12-03 08:40:00 +08:00
EXP 为model添加num_nodes 2025-12-03 08:40:00 +08:00
EXPB 为model添加num_nodes 2025-12-03 08:40:00 +08:00
GWN 适配GraphWaveNet 2025-12-03 12:05:02 +08:00
HI 更新iTransformer, HI配置。更新TS数据集载入方式 2025-12-10 11:01:00 +08:00
Informer 兼容InFormer 2025-12-11 23:16:25 +08:00
MTGNN impl mtgnn 2025-12-10 23:31:17 +08:00
MegaCRN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
NLT 为model添加num_nodes 2025-12-03 08:40:00 +08:00
PDG2SEQ 为model添加num_nodes 2025-12-03 08:40:00 +08:00
PatchTST 改进trainer,使用统一的loader而不是TSLoader,仅在Trainer上做了shape变换,确定尽可能少改动数据。 2025-12-14 17:47:38 +08:00
REPST 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STAEFormer 更新配置,METR-LA -STAEFormer 2025-12-03 18:51:54 +08:00
STAWnet 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STFGNN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGNCDE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGNRDE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STGODE 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STID 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STIDGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STMLP 为model添加num_nodes 2025-12-03 08:40:00 +08:00
STSGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
ST_SSL 为model添加num_nodes 2025-12-03 08:40:00 +08:00
TCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
TWDGCN 为model添加num_nodes 2025-12-03 08:40:00 +08:00
iTransformer 改进trainer,使用统一的loader而不是TSLoader,仅在Trainer上做了shape变换,确定尽可能少改动数据。 2025-12-14 17:47:38 +08:00
args_parser.py 简化trainer,修复设备bug,新增批量运行脚本 2025-12-10 21:08:20 +08:00