Skip to content

Latest commit

 

History

History
22 lines (12 loc) · 703 Bytes

README.md

File metadata and controls

22 lines (12 loc) · 703 Bytes

ST-ResNet in PyTorch

Demo code snippets (with dataset BikeNYC) of Deep Spatio-Temporal Residual Networks (ST-ResNet) from the paper "Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Prediction".

Prerequisites

  • Python 3.6
  • Pytorch 1.0

Usage

This is only demo code snippets for reference.

Reference

Zhang, Junbo, Yu Zheng, and Dekang Qi. "Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Prediction." AAAI. 2017. https://arxiv.org/pdf/1610.00081.pdf

Refered implementations

Keras: https://github.com/lucktroy/DeepST

Tensorflow: https://github.com/snehasinghania/STResNet