haeuuu/Graph-Convolutional-Matrix-Completion

pytorch implementation of GCMC (https://arxiv.org/pdf/1706.02263.pdf)

Graph-Convolutional-Matrix-Completion

• python implementation of GCMC
• paper : Graph Convolutional Matrix Completion

✅ ML-100k

Usage

python main.py train --n_layers=1

Results

# n_layers = 1

Best Valid RMSE : 0.9042
Best Test RMSE : 0.9255

# n_layers = 2

Best Valid RMSE : 0.9080
Best Test RMSE : 0.9282