TorchKGE：基于 Python 与 PyTorch 的知识图谱嵌入工具
计算与语言
2020-09-09 v1 人工智能
摘要
TorchKGE 是一个仅依赖 PyTorch 的 Python 知识图谱（KG）嵌入模块。该包为研究人员和工程师提供了简洁高效的 API 以设计和测试新模型。其具备 KG 数据结构、简易模型接口以及用于负采样和模型评估的模块。其主要优势是针对链接预测任务（KG 嵌入的核心应用）的极快评估模块。多种 KG 嵌入模型也已实现。我们特别关注代码效率与简洁性、文档以及 API 的一致性。该包以 BSD 许可证通过 PyPI 分发。源代码及文档与部署指引见 https://github.com/torchkge-team/torchkge。
引用
@article{arxiv.2009.02963,
title = {TorchKGE: Knowledge Graph Embedding in Python and PyTorch},
author = {Armand Boschin},
journal= {arXiv preprint arXiv:2009.02963},
year = {2020}
}
备注
Paper presented at KDD-IWKG 2020