CrickWu / GCMC

Code for Graph Convolutional Matrix Factorization for Bipartite Edge Prediction

Graph Convolutional Matrix Completion for Bipartite Edge Prediction

Paper

Graph Convolutional Matrix Completion for Bipartite Edge Prediction - Yuexin Wu, Hanxiao Liu and Yiming Yang KDIR 2018. (Best Student Paper)

The framework tries to solve the biparite edge prediction (BEP) problem via decomposing the middle edges as initial node vectors which are later passed through graph convolution neural networks for better hidden representations to the final prediction.

Dependencies

Python == 2.7, lasagne == 0.1, theano == 0.8, numpy, scipy, joblib, scikit-learn, nose-parameterized

Use Anaconda

If you have Anaconda, you can install the dependencies using the following command: conda create --name gcmc --file requirement.txt

How to Run

Preprocess

Uncompress data using 7z x data.7z.

Use GPU

If you have installed cuda support, add THEANO_FLAGS='device=cuda0,floatX=float32' before executing the commnad. In experiments, multi-core cpus are more preferable (with --n_jobs option).

Simple Example

python src/run.py --dataset cmu --layers 80 70 5 --iters 1000 --pos_up_ratio 10. --fold 1 --save_log --save_model --binary_graph

This will create logs and model files in save folder. The final result is saved in gcmc.log.

See python src/run.py --help for more option explanations.

Full Experiment in Paper

NOTICE: This may take LONG time. Consider running single experiments first to estimate time.

General --iters 1000 should be enough for all datasets. In our experiment, the exact parameters can be found in run_all.sh.

Execute the following command to run on all datasets.

bash run_all.sh

Citation

If you find the repository useful for your publication, please consider citing our paper:

@conference{wu2018graph,
author={Yuexin Wu and Hanxiao Liu and Yiming Yang},
title={Graph Convolutional Matrix Completion for Bipartite Edge Prediction},
booktitle={Proceedings of the 10th International Joint Conference on Knowledge Discovery, Knowledge Engineering and Knowledge Management - Volume 1: KDIR,},
year={2018},
pages={51-60},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0006900000510060},
isbn={978-989-758-330-8},
}

For any questions and comments, please send your email to [e-mail].