DGL-KE使用

DGL-KE是亚马逊开发的基于DGL 的知识图片嵌入库,提供了TransE,TransR等一系列嵌入方法,可以使用命令行的方式快速得到知识图谱中实体和关系的嵌入。

我遇到一个需求,需要将知识图谱中的实体和关系嵌入到低维向量,我首先想到的是使用清华的开源知识图谱嵌入库open-ke,但是当我在Ubuntu上面安装好时,我发现其只能支持在GPU上进行嵌入计算,但是我的虚拟机上没有条件安装CUDA,因此我将目光转向DGL-KE。

DGL-KE入门教程:
首先在Ubuntu上安装:

sudo pip3 install torch
sudo pip3 install dgl==0.4.3
sudo pip3 install dglke

然后我们来测试是否成功安装:
新建一个文件夹:

mkdir my_task 

猜你喜欢

转载自blog.csdn.net/cobracanary/article/details/119913870
今日推荐