基于注意力的学习嵌入在知识图中的关系预测 我们的论文的源代码:此出版物的博客 。 要求 请从上面的链接下载miniconda并使用以下命令创建环境: conda env create -f pytorch35.yml 在执行程序之前,请先激活环境,如下所示: source activate pytorch35 数据集 我们使用了五个不同的数据集来评估我们的模型。 下面列出了所有数据集及其文件夹名称。 Freebase:FB15k-237 词网:WN18RR 内尔:NELL-995 亲属关系 UMLS:umls 训练 参数: --data :指定数据集的文件夹名称。