hhh920406 / gcmc-2

PyTorch and PyTorch geometric based implementation of Graph Convolutional Matrix Completion.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Graph Convolutional Matrix Completion based on Pytorch

PyTorch and PyTorch geometric based implementation of Graph Convolutional Matrix Completion.

This repository is NOT an official implementation of that paper.
The official implementation is this which is based on Tensorflow.

Train and Test

Default training dataset is MovieLens-100k which will be automatically downloaded in ./data/ when running main.py.

Docker (Recommend)

  • Install docker and docker-compose (see docker official document)
  • Clone this repository
git clone https://github.com/tanimutomo/gcmc.git
  • Build the docker container
docker-compose -f ./docker/docker-compose-{cpu/gpu}.yml build
  • Train and test the model
docker-compose -f ./docker/docker-compose-{cpu/gpu}.yml run experiment python3 main.py

Local

Installation of Pytorch Geometric is difficult and can destroy your local python environment.
So, if you already installed docker and docker-compose in your machine, recommend to use Docker (above).
Please see Pytorch Geometirc document for more details.

Install requirements

  • torch==1.0.0
  • torchvision
  • comet_ml (you must have comet-ml account)
  • torch-scatter
  • torch-sparse
  • torch-cluster
  • torch-spline_conv
  • torch-geometric

About

PyTorch and PyTorch geometric based implementation of Graph Convolutional Matrix Completion.


Languages

Language:Python 93.0%Language:Dockerfile 7.0%