This is the code of STKGRec.
- python 3.8
- pytorch 1.7
python main.py --dataset nyc
Please cite as:
@article{STGRec,
author = {Wei Chen and Huaiyu Wan
and Shengnan Guo and Haoyu Huang
and Shaojie Zheng and Jiamu Li
and Shuohao Lin and Youfang Lin}
title = {Building and exploiting spatial–temporal knowledge graph for next POI recommendation},
journal = {Knowledge-Based Systems},
year = {2022}
}