/SiReN-reco

Primary LanguagePython

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