From 767950752307bb0d67315728ca7a9e7128010eba Mon Sep 17 00:00:00 2001 From: HengZhang Date: Mon, 3 Mar 2025 16:45:34 +0800 Subject: [PATCH] add url --- README.md | 28 ++++++++++++++++++++++++++++ Result.xlsx | Bin 12126 -> 12130 bytes 2 files changed, 28 insertions(+) diff --git a/README.md b/README.md index ddd33c2..1a4ea28 100644 --- a/README.md +++ b/README.md @@ -22,6 +22,8 @@ pip install -r requirements.txt pip install pyyaml tqdm statsmodels h5py kagglehub torch torchvision torchaudio torchdiffeq fastdtw notebook ``` + + # 快速开始 参考baseline.ipynb中的命令执行，或者使用下面的命令：（请确保当前目录为TrafficWheel） @@ -39,6 +41,8 @@ run.py会自动完成数据集下载、模型训练/评估工作。 :warning:现有的模型性能数据存放在[Result.xlsx](./Result.xlsx)，不必浪费资源再跑一遍。 + + # 测试模型 在实验结束后，模型的存档文件会被保存在 `experiments/dataset/训练时间 `文件夹下，共有4个文件。 @@ -52,6 +56,8 @@ run.py会自动完成数据集下载、模型训练/评估工作。 :warning:注意，请及时删除experiments文件夹中不必要的文件，要不整个文件夹会越堆越大。 + + # 更改配置 在config文件夹中，存放了每个模型的配置文件。每个数据集单独配置，使用yaml格式。 @@ -66,6 +72,8 @@ run.py会自动完成数据集下载、模型训练/评估工作。 一般不建议对基线参数进行修改，按默认跑是最稳定的。 + + # 开发模型 首先你需要创建一个开发分支dev，并切换到开发分支 @@ -91,6 +99,8 @@ git push origin dev 模型开发完成后，需要合并到main分支，在[这里](https://github.zhang-heng.com/czzhangheng/TrafficWheel/pulls)提交pull request。 + + # 已知问题 目前，实测以下模型性能与原报告相比指标偏高：ARIMA、TCN、DCRNN @@ -99,3 +109,21 @@ STGCN在载入图时会有未知warning 以下模型由于没有源码暂未实现：HA、VAR、FC-LSTM、GRU-ED + + +# 源代码及论文 + +| 论文 | 代码 | +| ------------------------------------------------------------ | ------------------------------------------------------------ | +| [HierAttnLSTM](https://arxiv.org/abs/2201.05760v4) | [代码](https://github.com/TeRyZh/Network-Level-Travel-Prediction-Hierarchical-Attention-LSTM) | +| [DSANET](https://dl.acm.org/doi/10.1145/3357384.3358132) | [代码](https://github.com/bighuang624/DSANet) | +| [STGCN](https://arxiv.org/abs/1709.04875) | [代码](https://github.com/hazdzz/STGCN) | +| [DCRNN](https://arxiv.org/abs/1707.01926) | [代码](https://github.com/chnsh/DCRNN_PyTorch) | +| [GraphWaveNet](https://arxiv.org/pdf/1906.00121.pdf) | [代码](https://github.com/SGT-LIM/GraphWavenet) | +| [STSGCN](https://aaai.org/ojs/index.php/AAAI/article/view/5438/5294) | [代码](https://github.com/SmallNana/STSGCN_Pytorch) | +| [AGCRN](https://arxiv.org/pdf/2007.02842) | [代码](https://github.com/LeiBAI/AGCRN) | +| [STFGNN](https://arxiv.org/abs/2012.09641) | [代码](https://github.com/lwm412/STFGNN-Pytorch) | +| [STGODE](https://arxiv.org/abs/2106.12931) | [代码](https://github.com/square-coder/STGODE) | +| [STG-NCDE](https://arxiv.org/abs/2112.03558) | [代码](https://github.com/jeongwhanchoi/STG-NCDE) | +| [DDGCRN](https://www.sciencedirect.com/science/article/abs/pii/S0031320323003710) | [代码](https://github.com/wengwenchao123/DDGCRN) | + diff --git a/Result.xlsx b/Result.xlsx index cd780095e431085120d51e5f04b148a01568f6b0..045e0986541856f49b01255975dbf794959ad41d 100644 GIT binary patch delta 1815 zcmV+y2k7|TUgBP`+6DxDU+eOd-Uc0iOK#gR5ZxE(9R%(biTc?VVh8><0n)-q^Vt{( zXo$8I0bJOI!KqgQ z*WOYYiv*W^rVaqwT9e+3AKMmx#!phP+RwqukUUG!8>VZxWhH`FvI@3X!43@DZ$`r) ztx78)B=41I{+64AI)^is(#y4U4N?@JNTw^5A|oYI4P&Ng+(r#B`I7D)KvbeEdL>K2 z>v*o~Agi;L^H_iZweO~iikv9g=eYu@ZK%IhRuy(X;UHq;jD9JZps;s;Ktd1>j3f%l zRE~*Cj8dd+)DI^Y5+a;LUvaZFsdXCJ(W%#36kuq{~shtBApQn zvan{lzJHz(AS}XKc4H+BxIY=3Lh3)08z=;nJKpTm6Ua1nuk%QR`1-CnYPj8C)Hhvc z)HJl*3aAO4RHSjj)Ee=3?~LWBZ+M+^8BozYp3ND$RJAm9s{ ztXW(2SDn8A0094!qyrI?=`0Sj6bdB*4uW(+I(P~I07EMP02lz1O)VOKoRmwC+AtJ_ z_m%n|Ebo)x5W)}KEdoMiQe0SD zCqZCX!6h}ms(A$~xzdtGr2PdiC?rM)ISp#A%N?}h86eG!!w@sxqU6*j~BZ(+OM-8tK-ouMb z^ZZ6h3Z|7&x%Gh@v_0P*e-car$~w~R`FmkV6llH|&cG`we5@sZGU{026*hbj7I!dr zT{_u`EXDtp^Wm7pi>0@WUm`DhjVInb4u@VeCyNmoPljY1zvJ076X38r71BIU@e4`5 zh65bTrcHo;3w|&k=E&0Z4_mWHB!&CQLDJ~4ZCPZtNO9!Hu71spT?xm-i0zD3@=v=?&Xl$ud}``pJe=3> z1AsRTAnE|I-(MbfHq|hm_ulJQuk3 z_U}7Bx!6a*hDSO8*V_1ewC^4R$i+c7Id=eN+$Jt)g8Lt{j1E5-1d7Q+6SL1ICjk)O zO2aS|#_t8+q2#_z|7NQrBz0ft_I1Dea082Tw6t_TM4McQ6ayB5++MZEP)AvbU`|J3IG5@lR7O|0@)Um@fH=6s4XfCjvW91cx*3oba`xL zlOQe?lk+Va0SA)~ED0m+l0E;r7PV0WmSV|v7a%KHSRR@+d2tF21xek{Pm)*1U!aLHhQ?|_9M9T;gC z(5V_z9b1J=+Nd8)E)--q3BQtNYf|}MO+IjaQcNf3l9~;11$!!)JTfHRcmF?Fl7^-t z4pd>ybbbE3B0yLMmF>n_T5vuZoI>e8(;E;1D`bOe_w4-r{ zHwWNj{A{RV_a-)o>ho8`- zOlT0>P92e`-A_{0ZaP_)knp!jH%S%?{ER1qBAz@Q3a-2Cj@-K zlND>HdaCmm0094!qyrJN1`0C*4xi;)gLDc206!}L02lz1Q7syOoRmwC+AtJ__m%n| zBJUI95W)}UeIgNM-buqNg!izOsh{5{ zNx`%-Dz^cUantkd@h5nKDeFkL=kJ9f5!U=5oPi&x@UfPE$f#q5AF$zru(*S{>(a?y zWGVW)oDat&UM&4({1W-mYdrDiaX9p&Ia!Rzcrqm8_#c{0GXV~}Q+b-_DS9Ev*KmOF zY+48Cm*7WpHTNuCf9EAe%D^cbGk3>H+B5N#ipmWH zPYU;QgOt%z)2_&Dk)kMwUHzIHyAqCvJhnGh$=}U7xlo#V@Tjhb@pxIk4*_ulJQud+_U}7BY3(Cm z!y_GlYfXHv?YqYS(mLoSmkz*;+r;5Ixc>pOkPbf>1eRgyTeH(9Cjk)AO2aS|hVKR6 zq2#_zyV^063MeCg|Ic^+b4WVhipqNfZOU3^C<;R4fn+r=8r^%^V6Ug#R| zBn+omfTf&Tid&$s1|m8jIUA7bRaYH=GYl07kk;TRi13ep*Mb)2&j+0{a*ERKl$+@g zcg*Ij7YFy&l!LKpnxKih7%qQ=pO&}x-Kc&k+g&k0dCD1K8mO&x{`v9#wcn?BRBG2( zQDc{G)2kBr)h>4y&pA{4TXL{2drt)JxiqBDepDXg>ziVM@<|vaemM2Rq=;rDx+L+% zbKCjkeVeWuFNj0`e~A-EjNK{1gg9t28xfCyyvu$1Bl2&P@d*{PTPalo1mtNybCa|z zfeD}GTZ428002LeJ}p-P)svPjKmm%A*DX2$1CtUiK>@>)R4z~f#gnBjIvbW@>RVL- p003SB000pH000000000000000y_5PbJ^@LSIWHpyA}jy^005@>Ts!~(