KnowledgeGraphEmbedding

RotatE: 复杂空间中的关系旋转知识图谱嵌入 简介

这是 PyTorch 实现的 RotatE 模型,用于知识图谱嵌入 (KGE)。我们提供了一套工具包,可实现多种流行的 KGE 模型的最先进性能。该工具包非常高效,能够在单个 GPU 上在几个小时内训练大型 KGE 模型。

• MRR, MR, HITS@1, HITS@3, HITS@10 (过滤后) - [x] AUC-PR (对于 Countries 数据集)

请参阅 codes/run.py 中的 argparse 配置,了解更多参数和详细信息。 测试 CUDA_VISIBLE_DEVICES=$GPU_DEVICE Python -u $CODE_PATH/run.py --do_test --cuda -init $SAVE 重现最佳结果 要在 ICLR 2019 论文 RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Space 中重现结果,您可以运行 best_config.sh 中的 bash 命令,以在五个广泛使用的数据集 (FB15k、FB15k-237、wn18、wn18rr 和 Countries) 上获得 RotatE、TransE 和 ComplEx 的最佳性能。 run.sh 脚本