AAAI 2021. GSNet: Learning Spatial-Temporal Correlations from Geographical and Semantic Aspects for Traffic Accident Risk Forecasting
train model on NYC:
python train.py --config config/nyc/GSNet_NYC_Config.json --gpus 0 train model on Chicago:
python train.py --config config/chicago/GSNet_Chicago_Config.json --gpus 0 The configuration file config.json contains three parts: Data, Training and Predict:
If you find this repository useful in your research, please cite the following paper:
@inproceedings{Wang2021gsnet, title={GSNet: Learning Spatial-Temporal Correlations from Geographical and Semantic Aspects for Traffic Accident Risk Forecasting}, author={Beibei Wang, Youfang Lin,Shengnan Guo, Huaiyu Wan}, booktitle={2021 AAAI Conference on Artificial Intelligence (AAAI'21)}, year={2021} }