/STTN

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

Primary LanguagePython

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