Implementation of the paper "Contrastive Learning with Bidirectional Transformers for Sequential Recommendation".
python main.py --template train_bert --dataset_code beauty
python main.py --template train_bert --dataset_code toys
python main.py --template train_bert --dataset_code ml-1m
python test.py --template test_bert
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.
Please cite our paper if you find our codes useful:
@inproceedings{CBiT,
author = {Hanwen Du and
Hui Shi and
Pengpeng Zhao and
Deqing Wang and
Victor S. Sheng and
Yanchi Liu and
Guanfeng Liu and
Lei Zhao},
title = {Contrastive Learning with Bidirectional Transformers for Sequential Recommendation},
booktitle = {CIKM},
year = {2022}
}