Awesome

ST-CGCN-PyTorch

Display of data sets and core code in STCGCN Paper: Spatial-Temporal Complex Graph Convolution Network for Traffic Flow Prediction Author: YX Bao, JS Huang, QQ Shen, et al.

Requirements

• PyTorch
• NumPy
• Matplotlib
• pandas
• numpy
• time
• os
• et al.

Example Dataset

PeMS03,PeMS04,PeMS07,PeMS08 and SZ-taxi (flow, speed, occupancy, weather)

Model

nets/ST-CGCN-main.py

This code mainly shows the fusion process of the complex correlation matrix.

Welcome to learn and put forward suggestions.