STTN

Spatial-Temporal Transformer算法复现时空transformer算法复现

DataSet：PeMSD7(M)选择前25个sensors

paper : <Spatial-Temporal Transformer Networks for Traffic Flow Forecasting>

pytorch 1.4.0

numpy 1.15.4

pandas 0.23.4

Name		Name	Last commit message	Last commit date
Latest commit History 28 Commits
Batch_Training_Version		Batch_Training_Version
PEMSD7		PEMSD7
GCN_models.py		GCN_models.py
One_hot_encoder.py		One_hot_encoder.py
README.md		README.md
ST_Transformer.py		ST_Transformer.py
layers.py		layers.py
train.py		train.py
validation.py		validation.py

Provide feedback