Home

Awesome

1.Recommended Environment:

numpy 1.18.1 tensorflow 1.14.0

2.Command for different datasets:

(1)PeMS04: python main.py --n_route 307 --file_path ./datasets/PeMS04/ --graph W_307.csv --feature V_307.csv --train_days 47 --validation_days 6 --test_days 6

(2)PeMS07(Default Dataset): python main.py --n_route 228 --file_path ./datasets/PeMS07/ --graph W_228.csv --feature V_228.csv --train_days 34 --validation_days 5 --test_days 5

(3)PeMS08: python main.py --n_route 170 --file_path ./datasets/PeMS08/ --graph W_170.csv --feature V_170.csv --train_days 50 --validation_days 6 --test_days 6

  1. In our project, we reference Bing Yu, Haoteng Yin and Zhanxing Zhu’s work STGCN.