SiReN
This is our PyTorch implementation code for our paper:
C. Seo et al., SiReN: Sign-Aware Recommendation Using Graph Neural Networks,
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