ST-TIS (Spatial-Temporal Transformer with Information Fusion and Region Sampling)

About

Implementation of the paper A Lightweight and Accurate Spatial-Temporal Transformer for Traffic Forecasting.

Thanks for your attention to this work. More details will be released soon.

Installation

Requirements

• Python 3.7 (Recommend Anaconda)
• pytorch 1.8.1

Description

• The data are available in the '/data', which is from the prior work STDN.
• Run with "python train.py" to train the model.