PyTorch implementation of Relational Link Prediction of RGCN (Modeling Relational Data with Graph Convolutional Networks). The code is sparsely optimized with torch_geometric library, which is builded based on PyTorch.
This is the original tensorflow implementation of link prediction of RGCN: https://github.com/MichSchli/RelationPrediction
Also, most of the functions in utils.py are brought from the following repository: https://github.com/dmlc/dgl/tree/master/examples/pytorch/rgcn, where this repositroy is based on another popular graph library, namely dgl.
This code is lastly tested with:
- python 3.7.x
- pytorch 1.7.x
- torch_geometric 1.7.x, with torch_scatter 2.0.6 and torch_sparse 0.6.9