This is an implementation for MQ-AttnSAS and MQ-AttnS3
modified based on (https://github.com/kang205/SASRec and https://github.com/pmixer/SASRec.pytorch), for baseline implementation, please check https://github.com/pmixer/SASRec.pytorch.
To verify the model result, just use
sh train.sh
modified based on (https://github.com/Li-fAngyU/sequential_rec), for baseline implementation, please check https://github.com/Li-fAngyU/sequential_rec and https://github.com/aHuiWang/CIKM2020-S3Rec.
To verify the pretrain procedure,
sh pretrain.sh
Or you can directly call,
sh finetune.sh
to reporduce the result