Title
Pykeen 1.0: A Python Library For Training And Evaluating Knowledge Graph Embeddings
Abstract
Recently, knowledge graph embeddings (KGEs) have received significant attention, and several software libraries have been developed for training and evaluation. While each of them addresses specific needs, we report on a community effort to a re-design and re-implementation of PyKEEN, one of the early KGE libraries. PyKEEN 1.0 enables users to compose knowledge graph embedding models based on a wide range of interaction models, training approaches, loss functions, and permits the explicit modeling of inverse relations. It allows users to measure each component's influence individually on the model's performance. Besides, an automatic memory optimization has been realized in order to optimally exploit the provided hardware. Through the integration of Optuna, extensive hyper-parameter optimization (HPO) functionalities are provided.
Year
DOI
Venue
2021
v22/20-825.html
JOURNAL OF MACHINE LEARNING RESEARCH
Keywords
DocType
Volume
Knowledge Graphs, Knowledge Graph Embeddings, Relational Learning
Journal
22
Issue
ISSN
Citations 
82
1532-4435
0
PageRank 
References 
Authors
0.34
0
7
Name
Order
Citations
PageRank
Mehdi Ali122.73
Max Berrendorf201.35
Charles Tapley Hoyt300.34
Laurent Vermue400.34
Sahand Sharifzadeh500.68
Volker Tresp62907373.75
Jens Lehmann75375355.08