STDN_PyTorch (Spatial-Temporal Dynamic Network)

About

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

Please refer to this repository for source code implemented with TensorFlow.

Installation

Requirements

• Python==3.6.5 (Recommend Anaconda)
• torch==1.3.1
• torchvision==0.4.2
• cudatoolkit=10.0

Steps

• pip install -r 'requirements.txt'

Usage

To be completed.