OvOYu/MVSTT

MVSTT

Python

MVSTT: A Multi-View Spatial-Temporal Transformer Network for Traffic Forecasting

Requirements

python 3.7
pytorch
numpy
pandas

Data Preparation

MVSTT is implemented on those several public traffic datasets.

PEMS03, PEMS04, PEMS07 and PEMS08 from STSGCN (AAAI-20).

Model Train

PEMS03, PEMS04, PEMS07, PEMS08:

python train.py --dataset PEMS08

Model Test

PEMS03, PEMS04, PEMS07, PEMS08:

python test.py --dataset PEMS08