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

Added PyTorch vs TF MAE comparison

This commit is contained in:
Chintan Shah 2019-11-03 20:14:40 -05:00
parent bb32eb0f46
commit a3b76a56c6
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@ -100,6 +100,16 @@ 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: