Awesome
STRNN
Predicting the Next Location: A Recurrent Model with Spatial and Temporal Contexts
Qiang Liu, Shu Wu, Liang Wang, Tieniu Tan (AAAI-16)
Performance of this implementation
(recall@1: 0.0017160630653176505)
(recall@5: 0.003539380072217654)
(recall@10: 0.004004147152407851)
(recall@100: 0.00464767080190197)
(recall@1000: 0.01394045534150613)
(recall@10000: 0.07866900175131349)
Requirements
1.python 3.5
2.pytorch 0.4.0
Usage
0. Preprocessing
If you use (prepro_xxx_50.txt
), you do not need to preprocess.
If you want to perform personally or modify the source,
Gowalla is required at ../dataset/
.
$ python preprocess.py
1. Training
$ python train_torch.py