/MASR

Code for Memory Bank Augmented Long-tail Sequential Recommendation

Primary LanguagePython

MASR: Memory Bank Augmented Long-tail Sequential Recommendation

This repository contains the source code of our paper, Memory Bank Augmented Long-tail Sequential Recommendation, which is accepted for publication at CIKM 2022. The implement of the sequence encoder follows BERT4Rec.

Requirements

  • Python 3.8.3
  • Pytorch 1.10.2
  • Transformers 4.15.0
  • wget
  • Tqdm

Training:

python main.py --template train_bert --use_memory_loss_tail -copy_tail_label