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

Velocity · 7d
+0.9
★ / day
Trend
→steady
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.