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