Skip to content
/ KGCL Public
forked from anonymousKGCL/KGCL

Pytorch implementation for paper: Knowledge Graph Contrastive Learning for Recommendation

Notifications You must be signed in to change notification settings

ymzhang6/KGCL

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

KGCL

This is the Pytorch implementation for the paper: Knowledge Graph Contrastive Learning for Recommendation, with CF learning part based on LightGCN

Enviroment Requirement

pip install -r requirements.txt

Dataset

We provide three processed datasets: Yelp2018 and Amazon-book and MIND.

An example to run KGCL

run KGCL on Yelp2018 dataset:

  • change base directory

Change ROOT_PATH in code/world.py

  • command

cd code && python main.py --dataset="yelp2018"

About

Pytorch implementation for paper: Knowledge Graph Contrastive Learning for Recommendation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%