Skip to content

JiachengLi1995/TiSASRec

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 

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 (j9li@eng.ucsd.edu).

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages