Skip to content

Latest commit

 

History

History
17 lines (14 loc) · 1.25 KB

README.md

File metadata and controls

17 lines (14 loc) · 1.25 KB

Implementation for the PREPRec paper, accepted at Recsys 2024. Our method enables cross-domain, cross-user zero-shot transfer competitive with in-domain SOTA models.

Quick start: Install packages from requirements.txt. Then follow the instructions in data folder for getting dataset and preprocessing. Then create a res folder to hold trained models and logs of results, and see sample.sh for examples for running and evaluating models.

Code credits: Original code is based off this pytorch SASRec implementation, with code also taken/repurposed from here, here, here and here.

Please cite our work if you use it:

@misc{wang2024pretrainedsequentialrecommendationframework,
      title={A Pre-trained Sequential Recommendation Framework: Popularity Dynamics for Zero-shot Transfer}, 
      author={Junting Wang and Praneet Rathi and Hari Sundaram},
      year={2024},
      eprint={2401.01497},
      archivePrefix={arXiv},
      primaryClass={cs.IR},
      url={https://arxiv.org/abs/2401.01497}, 
}