-
RGCN
-
..
-
README.md
-
generate_FB_BGPs.py
-
generate_WN_BGPs.py
-
generate_wikikg-v2-2015_BGPs.py
-
main.py
-
models.py
-
utils.py
README.md
PyTorch RGCN (Link Prediction)
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.
About
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.
Requirements
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