xueyagaga / STGCN-PyTorch Public

forked from Aguin/STGCN-PyTorch

Notifications You must be signed in to change notification settings
Fork 0
Star 0

PyTorch implementation of STGCN

0 stars 17 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
dataset		dataset
save		save
README.md		README.md
load_data.py		load_data.py
main.ipynb		main.ipynb
stgcn.py		stgcn.py
utils.py		utils.py

Repository files navigation

STGCN-PyTorch

Paper & Official Implementation

Yu, B., Yin, H., & Zhu, Z. (2017). Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting. arXiv preprint arXiv:1709.04875.

TF1 by VeritasYin

Alternative Implementations

MXNet by Davidham3

PyTorch by FelixOpolka

Results

Model		PeMSD7(M) 15min
	MAE	MAPE(%)	RMSE
STGCN(Cheb)	2.212	5.184	3.971

About

PyTorch implementation of STGCN

Report repository

Releases

No releases published

Packages

No packages published

Languages

Jupyter Notebook 66.1%
Python 33.9%