czzhangheng
/
DCRNN
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Moved comparison to the top

This commit is contained in:
Chintan Shah 2019-11-03 20:17:58 -05:00
parent a3b76a56c6
commit d92490b808
1 changed files with 21 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
README.md
View File

@ -7,23 +7,33 @@ Yaguang Li, Rose Yu, Cyrus Shahabi, Yan Liu, [Diffusion Convolutional Recurrent
## Requirements ## Requirements
torch * torch
scipy>=0.19.0 * scipy>=0.19.0
numpy>=1.12.1 * numpy>=1.12.1
pandas>=0.19.2 * pandas>=0.19.2
pyyaml * pyyaml
statsmodels * statsmodels
tensorflow>=1.3.0 * tensorflow>=1.3.0
torch * torch
tables * tables
future * future
Dependency can be installed using the following command: Dependency can be installed using the following command:
```bash ```bash
pip install -r requirements.txt pip install -r requirements.txt
``` ```
### Comparison with Tensorflow implementation
In MAE (For LA dataset, PEMS-BAY coming in a while)
| Horizon | Tensorflow | Pytorch |
|:--------|:--------:|:--------:|
| 1 Hour | 3.69 | 3.12 |
| 30 Min | 3.15 | 2.82 |
| 15 Min | 2.77 | 2.56 |
## Data Preparation ## Data Preparation
The traffic data files for Los Angeles (METR-LA) and the Bay Area (PEMS-BAY), i.e., `metr-la.h5` and `pems-bay.h5`, are available at [Google Drive](https://drive.google.com/open?id=10FOTa6HXPqX8Pf5WRoRwcFnW9BrNZEIX) or [Baidu Yun](https://pan.baidu.com/s/14Yy9isAIZYdU__OYEQGa_g), and should be The traffic data files for Los Angeles (METR-LA) and the Bay Area (PEMS-BAY), i.e., `metr-la.h5` and `pems-bay.h5`, are available at [Google Drive](https://drive.google.com/open?id=10FOTa6HXPqX8Pf5WRoRwcFnW9BrNZEIX) or [Baidu Yun](https://pan.baidu.com/s/14Yy9isAIZYdU__OYEQGa_g), and should be
put into the `data/` folder. put into the `data/` folder.
@ -100,16 +110,6 @@ python -m scripts.eval_baseline_methods --traffic_reading_filename=data/metr-la.
![PyTorch Results](figures/result4.png "PyTorch Results") ![PyTorch Results](figures/result4.png "PyTorch Results")
### Comparison with Tensorflow implementation
In MAE
| Horizon | Tensorflow | Pytorch |
|:--------|:--------:|:--------:|
| 1 Hour | 3.69 | 3.12 |
| 30 Min | 3.15 | 2.82 |
| 15 Min | 2.77 | 2.56 |
## Citation ## Citation
If you find this repository, e.g., the code and the datasets, useful in your research, please cite the following paper: If you find this repository, e.g., the code and the datasets, useful in your research, please cite the following paper: