HyperAIHyperAI

Command Palette

Search for a command to run...

PyKEEN 1.0:一个用于训练和评估知识图谱嵌入的 Python 库

Mehdi Ali Max Berrendorf Charles Tapley Hoyt Laurent Vermue Sahand Sharifzadeh Volker Tresp Jens Lehmann

摘要

近年来,知识图谱嵌入(Knowledge Graph Embeddings, KGEs)受到广泛关注,为此已开发出多个软件库以支持KGE的训练与评估。尽管各类工具各有侧重,我们仍通过社区协作对早期KGE库之一——PyKEEN进行了重新设计与重实现。PyKEEN 1.0 允许用户基于多种交互模型、训练方法、损失函数构建知识图谱嵌入模型(KGEMs),并支持显式建模逆关系。此外,系统实现了自动内存优化机制,以充分挖掘硬件资源的潜力;同时通过集成Optuna,提供了强大的自动超参数优化(Hyper-Parameter Optimization, HPO)功能。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供