论文标题
Torchkge:嵌入Python和Pytorch中的知识图
TorchKGE: Knowledge Graph Embedding in Python and PyTorch
论文作者
论文摘要
Torchkge是仅依赖Pytorch的知识图(kg)的Python模块。该软件包为研究人员和工程师提供了干净有效的API,以设计和测试新型号。它具有KG数据结构,简单的模型接口和模块,用于负采样和模型评估。它的主要优势是针对链接预测任务的非常快速的评估模块,这是KG嵌入的中心应用。也已经实施了各种千克嵌入模型。特别关注代码效率和简单性,文档和API一致性。它是在BSD许可下使用PYPI分发的。可以在https://github.com/torchkge-team/torchkge上找到文档和部署的源代码和指针。
TorchKGE is a Python module for knowledge graph (KG) embedding relying solely on PyTorch. This package provides researchers and engineers with a clean and efficient API to design and test new models. It features a KG data structure, simple model interfaces and modules for negative sampling and model evaluation. Its main strength is a very fast evaluation module for the link prediction task, a central application of KG embedding. Various KG embedding models are also already implemented. Special attention has been paid to code efficiency and simplicity, documentation and API consistency. It is distributed using PyPI under BSD license. Source code and pointers to documentation and deployment can be found at https://github.com/torchkge-team/torchkge.