OvOYu/STGHTN

STGHTN

Python

STGHTN:Spatial-Temporal Gated Hybrid Transformer Network for Traffic Flow Forecasting

Requirements

python 3.7
pytorch
numpy
pandas

Data Preparation

STGHTN 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