Polaris-JZ / DDRM

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Denoising Diffusion Recommender Model

This is the pytorch implementation of our paper

Denoising Diffusion Recommender Model

Environment

  • Anaconda 3
  • Python 3.8.12
  • Pytorch 1.7.0
  • Numpy 1.21.2

Training

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

About


Languages

Language:Python 100.0%