← all repositories

pykeen/pykeen

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

2k stars Python ML Frameworks
pykeen
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.