Skip to content

Re-Implement "Graph Convolutional Matrix Completion" (PyTorch and PyTorch Geometric)

Notifications You must be signed in to change notification settings

tanimutomo/gcmc

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Graph Convolutional Matrix Completion (Pytorch)

Re-implementation of Graph Convolutional Matrix Completion (PyTorch and PyTorch Geometric)

overview

approach

Note

This repository is NOT an official implementation of the paper.
The official implementation is this (Tensorflow).
Our experimental result is shown below and it doesn't reach to the score of the original.

Setup

  • Setup a virtual environment with python 3.6 or newer
  • Install requirements (pip)
    pip install -r requirements/1.txt
    pip install --verbose --no-cache-dir -r requirements/2.txt
    pip install -r requirements/3.txt
    

Installation of Pytorch Geometric is very troublesome and may destroy your python environment.
So, we strongly recommend to use a virtual environment (e.g. pyenv, virtualenv, pipenv, etc.).
Please see Pytorch Geometirc official document for more details.

Train and Test

cd src
python train.py
  • Configuration:
    The settings for train and test are in config.yml.

  • Dataset:
    Training dataset is MovieLens-100k. The dataset is automatically downloaded in data/ by running src/train.py.

Results

Note that this repo doesn't reach to the original one.

Test RMSE
Ours 0.968
Original 0.910

About

Re-Implement "Graph Convolutional Matrix Completion" (PyTorch and PyTorch Geometric)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages