This is the pytorch implementation of our paper
Denoising Diffusion Recommender Model
- Anaconda 3
- Python 3.8.12
- Pytorch 1.7.0
- Numpy 1.21.2
Run the MF-based DDRM on ML-1M dataset:
cd DDRM_MF
python train.py
Run the LightGCN-based DDRM on ML-1M dataset:
cd DDRM_LightGCN
python train.py
Run the SGL-based DDRM on ML-1M dataset:
cd DDRM_SGL
python train.py