ZJUDataIntelligence / HDM-GNN

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

1. cd to data folder: unzip data.zip
2. cd to model folder: python train.py --config ../config/config.json --output ../output/