
DCRNN: 基于Tensorflow的扩散卷积循环神经网络实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
DCRNN是一款基于TensorFlow框架开发的扩散卷积循环神经网络模型,适用于时空序列预测任务,尤其在交通流量预测中表现出色。
扩散卷积循环神经网络:数据驱动的交通预测
这是以下论文中Diffusion Convolutional Recurrent Neural Network(DCRNN)的TensorFlow实现:
Yaguang Li、Rose Yu、Cyrus Shahabi、Yan Liu,ICLR 2018。
要求如下:
- scipy>=0.19.0
- numpy>=1.12.1
- pandas>=0.19.2
- scikit-learn
- tensorflow>=1.3.0
可以使用以下命令安装依赖项:
```
pip install -r requirements.txt
```
数据准备:
洛杉矶(METR-LA)和湾区(PEMS-BAY)的交通数据文件,即metr-la.h5和pems-bay.h5需要放入data文件夹。*.h5 文件使用HDF5文件格式将数据存储在panads.DataFrame中。
下面是一个例子:
```
传感器_0
```
全部评论 (0)
还没有任何评论哟~


