ml-lab / repeatnet

Neural Attentive Recommendation Machine

This is our implementation for the paper:

Pengjie Ren, Jing Li, Zhumin Chen, Zhaochun Ren, Jun Ma and Maarten de Rijke (2019). RepeatNet: A Repeat Aware Neural Recommendation Machine for Session-based Recommendation. In Proceedings of The Thirty-Third AAAI Conference on Artificial Intelligence.

We have additionally released our Chainer implementation of RepeatNet. For YOOCHOOSE and DIGINETICA datasets, please refer to https://github.com/lijingsdu/sessionRec_NARM. For LASTFM dataset, please refer to the processing code in data/lastfm.

Please cite our AAAI'19 paper. Thanks!