Skip to content
/ HGARN Public

[TITS 2024] Activity-aware human mobility prediction with hierarchical graph attention recurrent network.

Notifications You must be signed in to change notification settings

YihongT/HGARN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HGARN

Source codes for Activity-aware human mobility prediction with hierarchical graph attention recurrent network., published in IEEE Transactions on Intelligent Transportation Systems (TITS) in 2024.

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{tang2024activity,
  title={Activity-aware human mobility prediction with hierarchical graph attention recurrent network},
  author={Tang, Yihong and He, Junlin and Zhao, Zhan},
  journal={IEEE Transactions on Intelligent Transportation Systems},
  year={2024},
  publisher={IEEE}
}

Acknowledgments

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