GitHub - wangxianggit/ST-GWCN: A traffic flow prediction model based on graph wavelet convolution network and gated temporal convolution network · GitHub

8000 GitHub - wangxianggit/ST-GWCN: A traffic flow prediction model based on graph wavelet convolution network and gated temporal convolution network · GitHub

Repository navigation

ST-GWCN

Public

About

A traffic flow prediction model based on graph wavelet convolution network and gated temporal convolution network

7 stars

1 watching

Report repository

Name		Name	Last commit message	Last commit date
config		config
figure		figure
model		model
script		script
README.md		README.md
data.csv		data.csv
lossquxian.py		lossquxian.py
main_road_traffic.py		main_road_traffic.py
requirements.txt		requirements.txt
test.py		test.py

Repository files navigation

Related works

TCN: An Empirical Evaluation of Generic Convolutional and Recurrent Networks for Sequence Modeling
GLU and GTU: Language Modeling with Gated Convolutional Networks
ChebyNet: Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering

Related code

TCN: https://github.com/locuslab/TCN
ChebyNet: https://github.com/mdeff/cnn_graph

Dataset

Source

METR-LA: DCRNN author's Google Drive
PEMS-BAY: DCRNN author's Google Drive
PeMSD7(M): STGCN author's GitHub repository

After prepare dataset, run: python main_road_traffic.py

About

A traffic flow prediction model based on graph wavelet convolution network and gated temporal convolution network

7 stars

1 watching

Report repository

Releases

Packages

Contributors

Languages

0