This is pytorch implementation for the spatial pre-trained model in our paper. The spatial pre-trained model is crucial for improving POI recommendation performance.
The code has been tested running under Python 3.6.5. The required packages are as follows:
- pytorch == 1.5.0
- python== 3.7.9