README.md

HGARN

Source codes for HGARN: Hierarchical Graph Attention Recurrent Network for Human Mobility Prediction.

Requirements

• python == 3.6
• torch == 1.7.0+cu110
• mpu == 0.23.1

See requirements.txt for more details.

Datasets

NYC and Tokyo Check-in Dataset.

Please refer to this repo.

Run

python train.py

Reference

Please cite our paper if you use the model in your own work:

@article{tang2022hgarn,
  title={HGARN: Hierarchical Graph Attention Recurrent Network for Human Mobility Prediction},
  author={Tang, Yihong and He, Junlin and Zhao, Zhan},
  journal={arXiv preprint arXiv:2210.07765},
  year={2022}
}

Acknowledgments

We refer to some of the data processing codes in this repo.