Skip to content

BorealisAI/sasrec-ccql

Repository files navigation

SASRec-CCQL

Transformer + RL based Recommender System : See paper Robust Reinforcement Learning Objectives for Sequential Recommender Systems

Arch

Installation

Use the install.sh script, give it executable permissions with chmod +x install.sh, and run it with ./install.sh. This uses conda.

Tested with the following lib versions:

python                    3.8.8
torch                     1.8.1+cu111
torch-geometric           2.2.0
torch-sparse              0.6.16
torchmetrics              0.3.2
torchvision               0.9.1+cu111
pandas                    1.5.2
numpy                     1.22.3
CUDA Version: 11.4

Running Experiments

python SASRec_ccql.py --dataset=RC15
python SASRec_ccql.py --dataset=Retailrocket

Results

Pytorch Baselines

RC15Top20

RetailrocketTop20

Paper Results

RRBaselines RRBaselines

Citation

@misc{mozifian2023robust,
      title={Robust Reinforcement Learning Objectives for Sequential Recommender Systems},
      author={Melissa Mozifian and Tristan Sylvain and Dave Evans and Lili Meng},
      year={2023},
      eprint={2305.18820},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}

About

PyTorch code of "Robust Reinforcement Learning Objectives for Sequential Recommender Systems"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published