# Pykg2vec: A Python Library for Knowledge Graph Embedding

arXiv: https://arxiv.org/abs/1906.04239 · HTML: https://mlsanity.com/paper/1906.04239/

## Key facts

- Cited by 9 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 4 Jun 2019.

## Built on by (most cited first, 9 of 9)

- [RS2G: Data-Driven Scene-Graph Extraction and Embedding for Robust Autonomous Perception and Scenario Understanding](https://mlsanity.com/paper/2304.08600/) (2023)
- [NeuralKG: An Open Source Library for Diverse Representation Learning of Knowledge Graphs](https://mlsanity.com/paper/2202.12571/) (2022)
- [μKG: A Library for Multi-source Knowledge Graph Embeddings and Applications](https://mlsanity.com/paper/2207.11442/) (2022)
- [A Novel Word Sense Disambiguation Approach Using WordNet Knowledge Graph](https://mlsanity.com/paper/2101.02875/) (2021)
- [DuetGraph: Coarse-to-Fine Knowledge Graph Reasoning with Dual-Pathway Global-Local Fusion](https://mlsanity.com/paper/2507.11229/) (2025)
- [Can Persistent Homology provide an efficient alternative for Evaluation of Knowledge Graph Completion Methods?](https://mlsanity.com/paper/2301.12929/) (2023)
- [NeuralKG-ind: A Python Library for Inductive Knowledge Graph Representation Learning](https://mlsanity.com/paper/2304.14678/) (2023)
- [Causal Discovery in Knowledge Graphs by Exploiting Asymmetric Properties of Non-Gaussian Distributions](https://mlsanity.com/paper/2106.01043/) (2021)
- [SparseTransX: Efficient Training of Translation-Based Knowledge Graph Embeddings Using Sparse Matrix Operations](https://mlsanity.com/paper/2502.16949/) (2025)

## Abstract (arXiv)

Pykg2vec is an open-source Python library for learning the representations of the entities and relations in knowledge graphs. Pykg2vec's flexible and modular software architecture currently implements 16 state-of-the-art knowledge graph embedding algorithms, and is designed to easily incorporate new algorithms. The goal of pykg2vec is to provide a practical and educational platform to accelerate research in knowledge graph representation learning. Pykg2vec is built on top of TensorFlow and Python's multiprocessing framework and provides modules for batch generation, Bayesian hyperparameter optimization, mean rank evaluation, embedding, and result visualization. Pykg2vec is released under the MIT License and is also available in the Python Package Index (PyPI). The source code of pykg2vec is available at https://github.com/Sujit-O/pykg2vec.