【ConvE巨无敌详细的复现】

ConvE巨无敌详细的复现

1、下载代码

代码链接

2、配置环境

这里没有特别要求torch版本和cuda版本，就根据电脑配置安装适用的版本即可

pytorch之前版本的链接

3、按照链接里的内容安装各种包，把报错都解决

很有用的参考链接

4、运行程序

不断运行该指令，如果报错就找上面的链接解决错误。

（如果电脑不能安装cuda，就不能用CUDA_VISIBLE_DEVICES=0跑了，代码里也没有不用gpu的选项，找个可以安装cuda的服务器跑吧，或者只能把代码改成cpu版本了）

CUDA_VISIBLE_DEVICES=0 python main.py --model conve --data FB15k-237 \
                                      --input-drop 0.2 --hidden-drop 0.3 --feat-drop 0.2 \
                                      --lr 0.003 --preprocess

notice：链接内说预处理只需要做一次，但如果不加–preprocess会报错

5、剩余的报错

No module named ‘xxxx’使用pip指令安装即可，如缺少nltk安装包

pip install nltk

如果有版本不匹配的，就根据报错中的需求版本，安装指定版本的工具包指令如下：

pip install nltk==3.8.1

参考链接：

(小白易懂)知识图谱嵌入论文复现：ConvE代码调试记录-CSDN博客

ConvE，知识图谱嵌入（KGE）论文复现（Ubuntu 20.04）-CSDN博客