├── .gitignore ├── .vscode └── launch.json ├── README.md ├── data_loader ├── __init__.py └── dataloader.py ├── dataset ├── PeMSD7_V_228.csv └── PeMSD7_W_228.csv ├── main.py ├── models ├── __init__.py ├── st_gat.py └── trainer.py ├── requirements.txt ├── runs ├── model_final_200epochs.pt └── model_final_60epochs.pt └── utils ├── __init__.py └── math_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/README.md -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/data_loader/__init__.py -------------------------------------------------------------------------------- /data_loader/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/data_loader/dataloader.py -------------------------------------------------------------------------------- /dataset/PeMSD7_V_228.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/dataset/PeMSD7_V_228.csv -------------------------------------------------------------------------------- /dataset/PeMSD7_W_228.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/dataset/PeMSD7_W_228.csv -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/st_gat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/models/st_gat.py -------------------------------------------------------------------------------- /models/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/models/trainer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/requirements.txt -------------------------------------------------------------------------------- /runs/model_final_200epochs.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/runs/model_final_200epochs.pt -------------------------------------------------------------------------------- /runs/model_final_60epochs.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/runs/model_final_60epochs.pt -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jswang/stgat_traffic_prediction/HEAD/utils/math_utils.py --------------------------------------------------------------------------------