The pytorch implementation of Traffic Flow Prediction via Spatial Temporal Graph Neural Network
pandas=1.0.5
, torch
, tqdm
, pytables
commands to install the above packages:
conda install pandas=1.0.5 tqdm pytables
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
input: 12 sampling (sampling frequency: 5 min), i.e., 1 hour
forecasting horizon: 12, i.e., 1 hour