ASTGCN-PyTorch

Rewrite the MXNet version ASTGCN using PyTorch. Please refer to the MXNet version for more details.

ATTENTION: This repo is still under active development.

The program is runnable, but the training result is NOT correct. Issues and/or pull requests are welcome.

Train model on PEMS04:

    python train.py --config configurations/PEMS04.conf --force True

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
configurations		configurations
data		data
lib		lib
model		model
test		test
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
requirements.txt		requirements.txt
train.py		train.py