Graph Convolutional Networks for Relational Link Prediction
This repository contains a TensorFlow implementation of Relational Graph Convolutional Networks (R-GCN), as well as experiments on relational link prediction. The description of the model and the results can be found in out paper:
Modeling Relational Data with Graph Convolutional Networks. Michael Schlichtkrull, Thomas N. Kipf, Peter Bloem, Rianne van den Berg, Ivan Titov, Max Welling (ArXiv 2017)
Requirements
- TensorFlow (>1.0)
Running demo
We provide a bash script to run a demo of our code. In the folder settings, a collection of configuration files can be found. The block diagonal model used in our paper is represented through the configuration file settings/gcn_block.exp. To run a given experiment, execute our bash script as follows:
bash run-train.sh \[configuration\]
We advise running the model on a modern GPU, as training can take up to several hours.
Citation
Please cite our paper if you use this code in your own work:
@article{schlichtkrull2017modeling,
title={Modeling Relational Data with Graph Convolutional Networks},
author={Schlichtkrull, Michael and Kipf, Thomas N and Bloem, Peter and Berg, Rianne van den and Titov, Ivan and Welling, Max},
journal={arXiv preprint arXiv:1703.06103},
year={2017}
}