Moved comparison to the top
This commit is contained in:
parent
a3b76a56c6
commit
d92490b808
42
README.md
42
README.md
|
|
@ -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.
|
|||
|
||||

|
||||
|
||||
### 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:
|
||||
|
|
|
|||
Loading…
Reference in New Issue