2022 Fall Semester KSE801 Team Project for team 5
Application of Spatio-Temporal GCN to predict passenger usage after subway line extension.
Source codes are based on PyTorch implementation of ST-GCN.
Passenger Usages of stations, Distance between stations
Run files in the following order.
python DataLoader.ipynb
python MatrixLoader.py
python main.py
python Baseline_comparison.py
This code is available at Python >= 3.6, but 3.8.15 is recommended.
To install required packages:
pip install -r requirements.txt