Skip to content

Latest commit

 

History

History
40 lines (34 loc) · 1.56 KB

README.md

File metadata and controls

40 lines (34 loc) · 1.56 KB
Raw

STDN (Spatial-Temporal Dynamic Network)

About

Source code of the paper Revisiting Spatial-Temporal Similarity: A Deep Learning Framework for Traffic Prediction

If you find this repository useful in your research, please cite the following paper:

@inproceedings{yao2019revisiting,
  title={Revisiting Spatial-Temporal Similarity: A Deep Learning Framework for Traffic Prediction},
  author={Yao, Huaxiu and Tang, Xianfeng and Wei, Hua and Zheng, Guanjie and Li, Zhenhui},
  booktitle={2019 AAAI Conference on Artificial Intelligence (AAAI'19)},
  year={2019} 
}

Installation

Requirements

  • Python 3.6 (Recommend Anaconda)
  • Ubuntu 16.04.3 LTS
  • Keras >= 2.0.8
  • tensorflow-gpu (or tensorflow) == 1.3.0 (install guide)

Usage

  • Download all codes (*.py) and put them in the same folder (let's name it "stdn") (stdn/*.py)
  • Create "data" folder in the same folder (stdn/data/)
  • Create "hdf5s" folder for logs (if not exist) (stdn/hdf5s/)
  • Download and extract all data files (*.npz) from data.zip and put them in "data" folder (stdn/data/*.npz)
  • Open terminal in the same folder (stdn/)
  • Run with "python main.py" for NYC taxi dataset, or "python main.py --dataset=bike" for NYC bike dataset
python main.py

python main.py --dataset=bike
  • Check the output results (RMSE and MAPE). Models are saved to "hdf5s" folder for further use.

Hyperparameters:

Please check the hyperparameters defined in main.py