MDSTN (Multi-Directional Short-Term Traffic Volume Prediction based on Spatio-Temporal Networks)

Prepare Data

• Step 1. Download the NYCBike dataset.
• Step 2. Process the POI data by getPOI.py, POIFromCSV.py in the folder of preprocess.
• Step 3. Process the traffic volume by ToMapNYCBike.py and ProcessInOutFlow.py.
• Step 4. Modify the file path in MDSTN.py.

Run MDSTN

python MDSTN.py

Dependencies

• Tensorflow > 2.0.0
• Anaconda 4.8.2 contains all the required packages.