HDM-GNN
HDM-GNN copied to clipboard
trafficstars
HDM-GNN
This is the Pytorch implementation of HDM-GNN in the paper: [HDM-GNN: A Heterogeneous Dynamic Multi-view Graph Neural Network for Crime Prediction] .
Requirements
- pytorch
- pytorch-geometric
- geopandas
Usage
To run the code
- cd to data folder:
unzip data.zip - cd to model folder:
python train.py --config ../config/config.json --output ../output/