PyKEEN 1.0：用于训练与评估知识图谱嵌入的 Python 库
机器学习
2020-07-31 v2 人工智能
机器学习
摘要
近来，知识图谱嵌入（KGEs）受到显著关注，并已开发若干软件库用于训练和评估 KGEs。尽管它们各自满足了特定需求，我们以一个社区项目重新设计并重新实现了最早的 KGE 库之一 PyKEEN。PyKEEN 1.0 使用户能够基于广泛的交互模型、训练方法、损失函数来组合知识图谱嵌入模型（KGEMs），并允许对逆关系进行显式建模。此外，已实现自动内存优化以最优地利用所提供的硬件，并通过集成 Optuna 提供了广泛超参数优化（HPO）功能。
引用
@article{arxiv.2007.14175,
title = {PyKEEN 1.0: A Python Library for Training and Evaluating Knowledge Graph Embeddings},
author = {Mehdi Ali and Max Berrendorf and Charles Tapley Hoyt and Laurent Vermue and Sahand Sharifzadeh and Volker Tresp and Jens Lehmann},
journal= {arXiv preprint arXiv:2007.14175},
year = {2020}
}