扩散卷积循环神经网络:数据驱动的交通预测
这是以下论文中Diffusion Convolutional Recurrent Neural Network的TensorFlow实现: Yaguang Li、Rose Yu、Cyrus Shahabi、Yan Liu,,ICLR 2018。
要求
scipy>=0.19.0
numpy>=1.12.1
熊猫>=0.19.2
皮亚尔
统计模型
张量流>=1.3.0
可以使用以下命令安装依赖项:
pip install -r requirements.txt
数据准备
洛杉矶(METR-LA)和湾区(PEMS-BAY)的交通数据文件,即metr-la.h5和pems-bay.h5 ,可以在或,需要放入data/文件夹。 *.h5文件使用HDF5文件格式将数据存储在panads.DataFrame 。 下面是一个例子:
传感器_0
传
1