Skip to content
/ STGCN Public
forked from Davidham3/STGCN

implementation of STGCN for traffic prediction in IJCAI2018

License

Notifications You must be signed in to change notification settings

lin-jj/STGCN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

STGCN

This is an implementation of Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting which has been accepted by IJCAI 2018.

requirements

mxnet >= 1.3.0

Dataset

dataset comes from PEMS, we sampled a little from Bay area. You can get sampled data from "data" folder.

The file "distance.csv" contains the distance between two stations, which we linked together.

The file "graph_signal_data_small.txt" contains the time series of each station, it's in a json format, you can use the function "data_preprocessing" in main.py to read it.

About

implementation of STGCN for traffic prediction in IJCAI2018

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%