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.
- PyTorch
- NumPy
- Matplotlib
- pandas
- numpy
- time
- os
- et al.
PeMS03,PeMS04,PeMS07,PeMS08 and SZ-taxi (flow, speed, occupancy, weather)
nets/ST-CGCN-main.py
This code mainly shows the fusion process of the complex correlation matrix.
Welcome to learn and put forward suggestions.