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.1 is recommended.
To install required packages:

pip install -r requirements.txt