/LSTPM

Ke Sun, Tieyun Qian, Tong Chen, Yile Liang, Quoc Viet Hung Nguyen, Hongzhi Yin: Where to Go Next: Modeling Long- and Short-Term User Preferences for Point-of-Interest Recommendation. AAAI 2020 accepted.

Primary LanguagePython

LSTPM

Implementation of paper "Ke Sun, Tieyun Qian, Tong Chen, Yile Liang, Quoc Viet Hung Nguyen, Hongzhi Yin. Where to Go Next: Modeling Long- and Short-Term User Preferences for Point-of-Interest Recommendation." accepted by AAAI 2020.

Update

  • The "distance.pkl" is generated by the "caculate_poi_distance" function in the "train.py".
  • We have uploaded the data processing script "data_process.py" for generating "foursquare_cut_one_day.pkl".

Requirements

  • python 3.6
  • pytorch 1.0.1