pykeen/pykeen
A Python library for training and evaluating knowledge graph embedding models with PyTorch.

Not currently ranked — collecting fresh signals.
star history
PyKEEN (Python Knowledge Embeddings) is a library for learning and evaluating knowledge graph embeddings, which are dense vector representations of entities and relations in a knowledge graph. It provides implementations of 40 embedding models and integrates with PyTorch Lightning and Optuna for hyperparameter optimization. The library supports link prediction and knowledge base completion tasks.
Frequently asked
- What is pykeen/pykeen?
- A Python library for training and evaluating knowledge graph embedding models with PyTorch.
- Is pykeen open source?
- Yes — pykeen/pykeen is open source, released under the MIT license.
- What language is pykeen written in?
- pykeen/pykeen is primarily written in Python.
- How popular is pykeen?
- pykeen/pykeen has 2k stars on GitHub.
- Where can I find pykeen?
- pykeen/pykeen is on GitHub at https://github.com/pykeen/pykeen.