huaishitou / tisasrec

TensorFlow implementation for paper Time Interval Aware Self-Attention for Sequential Recommendation.

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.

Datasets

This repo includes ml-1m dataset as an example.

For Amazon dataset, you could download Amazon review data from here..

Model Training

To train our model on ml-1m (with default hyper-parameters):

python main.py --dataset=ml-1m --train_dir=default 

Misc

The implemention of self attention is modified based on this.

Contact

If you have any questions, please send me an email ([email protected]).