PyTorch Lightning implementation of different knowledge graph embeddings
This repository aims to gather different knowledge graph embeddings, in order to compare them on different datasets.
This repository uses parts of the code of the following repositories:
- https://github.com/DeepGraphLearning/KnowledgeGraphEmbedding: datasets, and TransE, DistMult, ComplEx, RotatE and pRotatE models
- https://github.com/magnusja/ConvE: ConvE model