Entity Type Prediction with Relational Graph Convolutional Network (PyTorch)

This post proposes a Python setup for entity type prediction on heterogenous graphs, using the Relational Graph Convolutional Network (R-GCN). The setup uses the RGCNConv module from PyTorch. The code discussed in this post can be found on GitHub. Before we dive into the setup in Python, knowledge graphs and the R-GCN model will be explained.

Knowledge Graphs

A (knowledge) graph is a relational data representation, expressing relations between entities. The Resource Description Framework (RDF) is a common …

deep learning graph-convolution-network graph learning towards-data-science