TiSASRec: Time Interval Aware Self-Attention for Sequential Recommendation
This is our TensorFlow implementation for the paper:
Jiacheng Li, Yujie Wang, Julian McAuley (2020). Time Interval Aware Self-Attention for Sequential Recommendation. WSDM'20
We refer to the repo SASRec.
Please cite our paper if you use the code or datasets.
The code is tested under a Linux desktop (w/ GTX 1080 Ti GPU) with TensorFlow.
For Pytorch version of TiSASRec, please refer to repo.
This repo includes ml-1m dataset as an example.
For Amazon dataset, you could download Amazon review data from here..
To train our model on
ml-1m (with default hyper-parameters):
python main.py --dataset=ml-1m --train_dir=default
The implemention of self attention is modified based on this.
If you have any questions, please send me an email (firstname.lastname@example.org).