Skip to main content

Knowledge Graph embedding in Python and PyTorch.

Project description

logo torchkge Documentation Status Updates

TorchKGE: Knowledge Graph embedding in Python and Pytorch.

TorchKGE is a Python module for knowledge graph (KG) embedding relying solely on Pytorch. This package provides researchers and engineers with a clean and efficient API to design and test new models. It features a KG data structure, simple model interfaces and modules for negative sampling and model evaluation. Its main strength is a highly efficient evaluation module for the link prediction task, a central application of KG embedding. It has been observed to be up to five times faster than AmpliGraph and twenty-four times faster than OpenKE. Various KG embedding models are also already implemented. Special attention has been paid to code efficiency and simplicity, documentation and API consistency. It is distributed using PyPI under BSD license.


If you find this code useful in your research, please consider citing our paper (presented at IWKG-KDD 2020):

    title={TorchKGE: Knowledge Graph Embedding in Python and PyTorch},
    author={Armand Boschin},
    booktitle={International Workshop on Knowledge Graph: Mining Knowledge Graph for Deep Insights},

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for torchkge, version 0.16.17
Filename, size File type Python version Upload date Hashes
Filename, size torchkge-0.16.17-py2.py3-none-any.whl (47.7 kB) File type Wheel Python version py2.py3 Upload date Hashes View
Filename, size torchkge-0.16.17.tar.gz (35.5 kB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page