This is the implementation of the paper "Diffusion Augmentation for Sequential Recommendation".
You can implement our model according to the following steps:
-
The handle dataset should be put into
./data/<dataset>/handled/
-
Install the necessary packages. Run the command:
pip install -r requirements.txt
-
To get train the DiffuASR and augment the corresponding dataset, please run the command:
bash ./experiments/diffusion.bash
-
Finally, you can run the following bash to test the augmention performance for Bert4Rec:
bash ./experiments/bert4rec.bash