Skip to content

Polaris-JZ/DDRM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages