├── .gitignore ├── LICENSE ├── README.md ├── data ├── metr-la │ ├── SE(METR-LA).txt │ ├── distances.csv │ ├── graph_sensor_locations.csv │ ├── metr-la.h5 │ └── pretrained │ │ ├── GMAN_METR-LA.data-00000-of-00001 │ │ ├── GMAN_METR-LA.index │ │ ├── GMAN_METR-LA.meta │ │ ├── dcrnn_config.yaml │ │ ├── dcrnn_pytorch_28.tar │ │ ├── dcrnn_test_pytorch.yaml │ │ ├── fc_lstm.pth │ │ └── graph_wavenet_repr.pth ├── model │ ├── dcrnn_bay.yaml │ ├── dcrnn_la.yaml │ ├── dcrnn_test_config.yaml │ └── pretrained │ │ ├── METR-LA │ │ ├── config.yaml │ │ ├── models-2.7422-24375.data-00000-of-00001 │ │ └── models-2.7422-24375.index │ │ └── PEMS-BAY │ │ ├── config.yaml │ │ ├── events.out.tfevents.1547170277.kakarot │ │ ├── models-1.6139-30780.data-00000-of-00001 │ │ └── models-1.6139-30780.index └── sensor_graph │ ├── adj_mx.pkl │ ├── adj_mx_bay.pkl │ ├── distances_la_2012.csv │ ├── graph_sensor_ids.txt │ └── graph_sensor_locations.csv ├── dcrnn_test.py ├── dcrnn_test_pytorch.py ├── dcrnn_train.py ├── dcrnn_train_pytorch.py ├── figures ├── dcrnn_architecture.jpg ├── gwnet_architecture.png ├── reproduced_papers_results.png ├── robustness_of_sensor_1.gif └── trafficprediction.png ├── gman_train.py ├── gwnet_test.py ├── gwnet_train.py ├── gwnet_viz.py ├── lib ├── AMSGrad.py ├── dataloaders │ └── dataloader.py ├── evaluate.py ├── gman_utils.py ├── logger.py ├── metrics │ ├── metrics_np.py │ ├── metrics_tf.py │ ├── metrics_torch.py │ └── test_metrics.py └── utils.py ├── model ├── pytorch │ ├── dcrnn_cell.py │ ├── dcrnn_model.py │ ├── dcrnn_supervisor.py │ ├── engine.py │ ├── gwnet_model.py │ ├── lstm_model.py │ └── supervisor.py └── tf │ ├── dcrnn_cell.py │ ├── dcrnn_model.py │ ├── dcrnn_supervisor.py │ ├── gman_model.py │ └── layers_tf.py ├── requirements.txt └── scripts ├── eval_baseline_methods.py ├── gen_adj_mx.py ├── generateSE.py └── node2vec.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/README.md -------------------------------------------------------------------------------- /data/metr-la/SE(METR-LA).txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/SE(METR-LA).txt -------------------------------------------------------------------------------- /data/metr-la/distances.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/distances.csv -------------------------------------------------------------------------------- /data/metr-la/graph_sensor_locations.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/graph_sensor_locations.csv -------------------------------------------------------------------------------- /data/metr-la/metr-la.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/metr-la.h5 -------------------------------------------------------------------------------- /data/metr-la/pretrained/GMAN_METR-LA.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/pretrained/GMAN_METR-LA.data-00000-of-00001 -------------------------------------------------------------------------------- /data/metr-la/pretrained/GMAN_METR-LA.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/pretrained/GMAN_METR-LA.index -------------------------------------------------------------------------------- /data/metr-la/pretrained/GMAN_METR-LA.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/pretrained/GMAN_METR-LA.meta -------------------------------------------------------------------------------- /data/metr-la/pretrained/dcrnn_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/pretrained/dcrnn_config.yaml -------------------------------------------------------------------------------- /data/metr-la/pretrained/dcrnn_pytorch_28.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/pretrained/dcrnn_pytorch_28.tar -------------------------------------------------------------------------------- /data/metr-la/pretrained/dcrnn_test_pytorch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/pretrained/dcrnn_test_pytorch.yaml -------------------------------------------------------------------------------- /data/metr-la/pretrained/fc_lstm.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/pretrained/fc_lstm.pth -------------------------------------------------------------------------------- /data/metr-la/pretrained/graph_wavenet_repr.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/metr-la/pretrained/graph_wavenet_repr.pth -------------------------------------------------------------------------------- /data/model/dcrnn_bay.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/dcrnn_bay.yaml -------------------------------------------------------------------------------- /data/model/dcrnn_la.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/dcrnn_la.yaml -------------------------------------------------------------------------------- /data/model/dcrnn_test_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/dcrnn_test_config.yaml -------------------------------------------------------------------------------- /data/model/pretrained/METR-LA/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/pretrained/METR-LA/config.yaml -------------------------------------------------------------------------------- /data/model/pretrained/METR-LA/models-2.7422-24375.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/pretrained/METR-LA/models-2.7422-24375.data-00000-of-00001 -------------------------------------------------------------------------------- /data/model/pretrained/METR-LA/models-2.7422-24375.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/pretrained/METR-LA/models-2.7422-24375.index -------------------------------------------------------------------------------- /data/model/pretrained/PEMS-BAY/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/pretrained/PEMS-BAY/config.yaml -------------------------------------------------------------------------------- /data/model/pretrained/PEMS-BAY/events.out.tfevents.1547170277.kakarot: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/pretrained/PEMS-BAY/events.out.tfevents.1547170277.kakarot -------------------------------------------------------------------------------- /data/model/pretrained/PEMS-BAY/models-1.6139-30780.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/pretrained/PEMS-BAY/models-1.6139-30780.data-00000-of-00001 -------------------------------------------------------------------------------- /data/model/pretrained/PEMS-BAY/models-1.6139-30780.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/model/pretrained/PEMS-BAY/models-1.6139-30780.index -------------------------------------------------------------------------------- /data/sensor_graph/adj_mx.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/sensor_graph/adj_mx.pkl -------------------------------------------------------------------------------- /data/sensor_graph/adj_mx_bay.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/sensor_graph/adj_mx_bay.pkl -------------------------------------------------------------------------------- /data/sensor_graph/distances_la_2012.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/sensor_graph/distances_la_2012.csv -------------------------------------------------------------------------------- /data/sensor_graph/graph_sensor_ids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/sensor_graph/graph_sensor_ids.txt -------------------------------------------------------------------------------- /data/sensor_graph/graph_sensor_locations.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/data/sensor_graph/graph_sensor_locations.csv -------------------------------------------------------------------------------- /dcrnn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/dcrnn_test.py -------------------------------------------------------------------------------- /dcrnn_test_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/dcrnn_test_pytorch.py -------------------------------------------------------------------------------- /dcrnn_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/dcrnn_train.py -------------------------------------------------------------------------------- /dcrnn_train_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/dcrnn_train_pytorch.py -------------------------------------------------------------------------------- /figures/dcrnn_architecture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/figures/dcrnn_architecture.jpg -------------------------------------------------------------------------------- /figures/gwnet_architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/figures/gwnet_architecture.png -------------------------------------------------------------------------------- /figures/reproduced_papers_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/figures/reproduced_papers_results.png -------------------------------------------------------------------------------- /figures/robustness_of_sensor_1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/figures/robustness_of_sensor_1.gif -------------------------------------------------------------------------------- /figures/trafficprediction.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/figures/trafficprediction.png -------------------------------------------------------------------------------- /gman_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/gman_train.py -------------------------------------------------------------------------------- /gwnet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/gwnet_test.py -------------------------------------------------------------------------------- /gwnet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/gwnet_train.py -------------------------------------------------------------------------------- /gwnet_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/gwnet_viz.py -------------------------------------------------------------------------------- /lib/AMSGrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/AMSGrad.py -------------------------------------------------------------------------------- /lib/dataloaders/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/dataloaders/dataloader.py -------------------------------------------------------------------------------- /lib/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/evaluate.py -------------------------------------------------------------------------------- /lib/gman_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/gman_utils.py -------------------------------------------------------------------------------- /lib/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/logger.py -------------------------------------------------------------------------------- /lib/metrics/metrics_np.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/metrics/metrics_np.py -------------------------------------------------------------------------------- /lib/metrics/metrics_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/metrics/metrics_tf.py -------------------------------------------------------------------------------- /lib/metrics/metrics_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/metrics/metrics_torch.py -------------------------------------------------------------------------------- /lib/metrics/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/metrics/test_metrics.py -------------------------------------------------------------------------------- /lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/lib/utils.py -------------------------------------------------------------------------------- /model/pytorch/dcrnn_cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/pytorch/dcrnn_cell.py -------------------------------------------------------------------------------- /model/pytorch/dcrnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/pytorch/dcrnn_model.py -------------------------------------------------------------------------------- /model/pytorch/dcrnn_supervisor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/pytorch/dcrnn_supervisor.py -------------------------------------------------------------------------------- /model/pytorch/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/pytorch/engine.py -------------------------------------------------------------------------------- /model/pytorch/gwnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/pytorch/gwnet_model.py -------------------------------------------------------------------------------- /model/pytorch/lstm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/pytorch/lstm_model.py -------------------------------------------------------------------------------- /model/pytorch/supervisor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/pytorch/supervisor.py -------------------------------------------------------------------------------- /model/tf/dcrnn_cell.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/tf/dcrnn_cell.py -------------------------------------------------------------------------------- /model/tf/dcrnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/tf/dcrnn_model.py -------------------------------------------------------------------------------- /model/tf/dcrnn_supervisor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/tf/dcrnn_supervisor.py -------------------------------------------------------------------------------- /model/tf/gman_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/tf/gman_model.py -------------------------------------------------------------------------------- /model/tf/layers_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/model/tf/layers_tf.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/eval_baseline_methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/scripts/eval_baseline_methods.py -------------------------------------------------------------------------------- /scripts/gen_adj_mx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/scripts/gen_adj_mx.py -------------------------------------------------------------------------------- /scripts/generateSE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/scripts/generateSE.py -------------------------------------------------------------------------------- /scripts/node2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tijsmaas/TrafficPrediction/HEAD/scripts/node2vec.py --------------------------------------------------------------------------------