ACRec

Implementation of the paper "Adpative Contrastive Learning for Sequential Recommendation".

Run beauty

python main.py --template train_bert --dataset_code beauty

Run Automotive

python main.py --template train_bert --dataset_code Automotive

Run grocery

python main.py --template train_bert --dataset_code grocery

Run Video_Games

python main.py --template train_bert --dataset_code Video_Games

Test a pretrained checkpoint on beauty

python test.py --template test_bert --dataset_code beauty

• The checkpoint file is stored at /experiments/test_2022-04-26_0/models/best_acc_model.pth . You may need to download the checkpoint file manually from git LFS.

Acknowledgements

Training pipeline is implemented based on this repo https://github.com/jaywonchung/BERT4Rec-VAE-Pytorch . We would like to thank the contributors for their work.