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

View File

@ -7,23 +7,33 @@ Yaguang Li, Rose Yu, Cyrus Shahabi, Yan Liu, [Diffusion Convolutional Recurrent
## Requirements
torch
scipy>=0.19.0
numpy>=1.12.1
pandas>=0.19.2
pyyaml
statsmodels
tensorflow>=1.3.0
torch
tables
future
* torch
* scipy>=0.19.0
* numpy>=1.12.1
* pandas>=0.19.2
* pyyaml
* statsmodels
* tensorflow>=1.3.0
* torch
* tables
* future
Dependency can be installed using the following command:
```bash
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
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.
@ -100,16 +110,6 @@ python -m scripts.eval_baseline_methods --traffic_reading_filename=data/metr-la.
![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
If you find this repository, e.g., the code and the datasets, useful in your research, please cite the following paper: