← all repositories

pykeen/pykeen

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

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

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