torchkge

TorchKGE: Knowledge Graph embedding in Python and PyTorch.

Showing:

Popularity

Downloads/wk

0

GitHub Stars

220

Maintenance

Last Commit

4mos ago

Contributors

8

Package

Dependencies

3

License

BSD license

Categories

Readme

========

TorchKGE

.. image:: https://graphs.telecom-paristech.fr/images/logo_torchKGE_small.png :align: right :width: 100px :alt: logo torchkge

.. image:: https://img.shields.io/pypi/v/torchkge.svg :target: https://pypi.python.org/pypi/torchkge

.. image:: https://travis-ci.com/torchkge-team/torchkge.svg?branch=master :target: https://travis-ci.com/torchkge-team/torchkge

.. image:: https://readthedocs.org/projects/torchkge/badge/?version=latest :target: https://torchkge.readthedocs.io/en/latest/?badge=latest :alt: Documentation Status

.. image:: https://pyup.io/repos/github/torchkge-team/torchkge/shield.svg :target: https://pyup.io/repos/github/torchkge-team/torchkge/ :alt: 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 <https://torchkge.readthedocs.io/en/latest/reference/evaluation.html> to be up to five times faster than AmpliGraph <https://docs.ampligraph.org/> and twenty-four times faster than OpenKE <https://github.com/thunlp/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.

Citations

If you find this code useful in your research, please consider citing our paper <https://arxiv.org/abs/2009.02963> (presented at IWKG-KDD <https://suitclub.ischool.utexas.edu/IWKG_KDD2020/index.html> 2020):

.. code-block::

@inproceedings{arm2020torchkge,
    title={TorchKGE: Knowledge Graph Embedding in Python and PyTorch},
    author={Armand Boschin},
    year={2020},
    month={Aug},
    booktitle={International Workshop on Knowledge Graph: Mining Knowledge Graph for Deep Insights},
}

Rate & Review

Great Documentation0
Easy to Use0
Performant0
Highly Customizable0
Bleeding Edge0
Responsive Maintainers0
Poor Documentation0
Hard to Use0
Slow0
Buggy0
Abandoned0
Unwelcoming Community0
100
No reviews found
Be the first to rate

Alternatives

No alternatives found

Tutorials

No tutorials found
Add a tutorial