/STKGRec

Primary LanguagePython

STKGRec

This is the pytorch implementation for our KBS paper: Building and exploiting spatial–temporal knowledge graph for next POI recommendation.

Requirements

  • python 3.8
  • pytorch 1.7

Run

python main.py --dataset nyc

Cite

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}
}