A PaddlePaddle implementation of Time Interval Aware Self-Attentive Sequential Recommendation.
论文：Time Interval Aware Self-Attentive Sequential Recommendation
| Datasets | Metrics | Paper's | Ours | abs. improv. |
|---|---|---|---|---|
| MovieLens-1m | HIT@10 | 0.8038 | 0.8050 | 0.0012 |
| MovieLens-1m | NDCG@10 | 0.5706 | 0.5752 | 0.0046 |
- Python >= 3
- PaddlePaddle >= 2.0.0
- see
requirements.txt
MovieLens-1m (max_len = 50)
bash ./script/train.sh模型在 200 epochs 左右收敛，日志见 nohup.out。
bash ./script/eval.sh可以得到如下结果：
@inproceedings{li2020time,
title={Time Interval Aware Self-Attention for Sequential Recommendation},
author={Li, Jiacheng and Wang, Yujie and McAuley, Julian},
booktitle={Proceedings of the 13th International Conference on Web Search and Data Mining},
pages={322--330},
year={2020}
}