├── .gitignore ├── DCRNN_baseline ├── gen_adj_mx.py ├── generate_training_data.py ├── speed_h5py.py └── success ├── DiDi_data ├── README.md └── didi_data_completion.ipynb ├── METR-LA ├── GCN+GRU+TIME.py ├── METR-LA_model.ipynb ├── README.md ├── cluster_gcn.py ├── generate_data.py ├── origin_data │ └── graph_sensor_ids.txt ├── train_model.py └── train_model_2.py ├── PeMSD7_data ├── PeMSD7_roadmap.ipynb └── README.md ├── README.md ├── STGCN_baseline ├── data │ ├── DATA-README.md │ └── METR-LA.zip ├── main.py ├── stgcn.py ├── tester.py └── utils.py ├── beijing_traffic ├── 01_road_feature_matrix_full.ipynb ├── 02_select_node_from_LargeMap.ipynb ├── 03_select_road_map_feature.ipynb ├── 04_select_speed_data.ipynb ├── 05_generate_training_data.ipynb ├── 06_seq2seq.py ├── 07_Fast_Unfolding.py ├── README.md ├── figure │ ├── baidu_data.PNG │ └── baidu_data_2.PNG ├── neighbours_1km.txt └── simpleRNN.py ├── data_processing └── road_map.ipynb ├── templates ├── seq2seq.ipynb ├── simpleRNN.ipynb ├── test.py └── time_series.ipynb └── thoughts └── README.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/.gitignore -------------------------------------------------------------------------------- /DCRNN_baseline/gen_adj_mx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/DCRNN_baseline/gen_adj_mx.py -------------------------------------------------------------------------------- /DCRNN_baseline/generate_training_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/DCRNN_baseline/generate_training_data.py -------------------------------------------------------------------------------- /DCRNN_baseline/speed_h5py.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/DCRNN_baseline/speed_h5py.py -------------------------------------------------------------------------------- /DCRNN_baseline/success: -------------------------------------------------------------------------------- 1 | success！ -------------------------------------------------------------------------------- /DiDi_data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/DiDi_data/README.md -------------------------------------------------------------------------------- /DiDi_data/didi_data_completion.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/DiDi_data/didi_data_completion.ipynb -------------------------------------------------------------------------------- /METR-LA/GCN+GRU+TIME.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/METR-LA/GCN+GRU+TIME.py -------------------------------------------------------------------------------- /METR-LA/METR-LA_model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/METR-LA/METR-LA_model.ipynb -------------------------------------------------------------------------------- /METR-LA/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/METR-LA/README.md -------------------------------------------------------------------------------- /METR-LA/cluster_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/METR-LA/cluster_gcn.py -------------------------------------------------------------------------------- /METR-LA/generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/METR-LA/generate_data.py -------------------------------------------------------------------------------- /METR-LA/origin_data/graph_sensor_ids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/METR-LA/origin_data/graph_sensor_ids.txt -------------------------------------------------------------------------------- /METR-LA/train_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/METR-LA/train_model.py -------------------------------------------------------------------------------- /METR-LA/train_model_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/METR-LA/train_model_2.py -------------------------------------------------------------------------------- /PeMSD7_data/PeMSD7_roadmap.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/PeMSD7_data/PeMSD7_roadmap.ipynb -------------------------------------------------------------------------------- /PeMSD7_data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/PeMSD7_data/README.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/README.md -------------------------------------------------------------------------------- /STGCN_baseline/data/DATA-README.md: -------------------------------------------------------------------------------- 1 | Original data from https://github.com/liyaguang/DCRNN 2 | -------------------------------------------------------------------------------- /STGCN_baseline/data/METR-LA.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/STGCN_baseline/data/METR-LA.zip -------------------------------------------------------------------------------- /STGCN_baseline/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/STGCN_baseline/main.py -------------------------------------------------------------------------------- /STGCN_baseline/stgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/STGCN_baseline/stgcn.py -------------------------------------------------------------------------------- /STGCN_baseline/tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/STGCN_baseline/tester.py -------------------------------------------------------------------------------- /STGCN_baseline/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/STGCN_baseline/utils.py -------------------------------------------------------------------------------- /beijing_traffic/01_road_feature_matrix_full.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/01_road_feature_matrix_full.ipynb -------------------------------------------------------------------------------- /beijing_traffic/02_select_node_from_LargeMap.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/02_select_node_from_LargeMap.ipynb -------------------------------------------------------------------------------- /beijing_traffic/03_select_road_map_feature.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/03_select_road_map_feature.ipynb -------------------------------------------------------------------------------- /beijing_traffic/04_select_speed_data.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/04_select_speed_data.ipynb -------------------------------------------------------------------------------- /beijing_traffic/05_generate_training_data.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/05_generate_training_data.ipynb -------------------------------------------------------------------------------- /beijing_traffic/06_seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/06_seq2seq.py -------------------------------------------------------------------------------- /beijing_traffic/07_Fast_Unfolding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/07_Fast_Unfolding.py -------------------------------------------------------------------------------- /beijing_traffic/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/README.md -------------------------------------------------------------------------------- /beijing_traffic/figure/baidu_data.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/figure/baidu_data.PNG -------------------------------------------------------------------------------- /beijing_traffic/figure/baidu_data_2.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/figure/baidu_data_2.PNG -------------------------------------------------------------------------------- /beijing_traffic/neighbours_1km.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/beijing_traffic/neighbours_1km.txt -------------------------------------------------------------------------------- /beijing_traffic/simpleRNN.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_processing/road_map.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/data_processing/road_map.ipynb -------------------------------------------------------------------------------- /templates/seq2seq.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/templates/seq2seq.ipynb -------------------------------------------------------------------------------- /templates/simpleRNN.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/templates/simpleRNN.ipynb -------------------------------------------------------------------------------- /templates/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/templates/test.py -------------------------------------------------------------------------------- /templates/time_series.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/templates/time_series.ipynb -------------------------------------------------------------------------------- /thoughts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RobinLu1209/STGCN/HEAD/thoughts/README.md --------------------------------------------------------------------------------