Skip to content

cw-seo/SiReN-reco

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SiReN

This is our PyTorch implementation code for our paper:

C. Seo et al., SiReN: Sign-Aware Recommendation Using Graph Neural Networks,

Paper in arXiv

Note. In this implementation, we use LightGCN (X. He et al., SIGIR'20) as the GNN model for the graph with positive edges in our SiReN method.

Environment Requirements

The code has been tested under Python 3.7.7. The required packages are as follows:

  • Pytorch == 1.5.0
  • Pytorch Geometric == 1.6.3

Example : ML-1M dataset

python main.py --dataset ML-1M --version 1 

Example : Amazon-Book dataset

python main.py --dataset amazon --version 1 

Example : Yelp dataset

python main.py --dataset yelp --version 1 

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages