OpenKE复现与使用(一)配置环境与复现TransE、ComplEx

0.注意点: 

                (1)一定要配置双环境,虚拟环境不能用GPU训练。

                (2)Windows系统openke可以运行,但是很难调,远没有直接配置一个双环境来的简单

                (3)参考链接里没有创建虚拟环境,但是我感觉还是创建一个虚拟环境,把项目分开比较好

                (4)openke的github里有快速开始文档,里面的第三条指令只是在做准备工作,第一次运行一遍就行了,第四条指令是把examples里的文件复制到OpenKE文件夹里,可以手动复制

1.安装双系统Ubuntu22.04,让OpenKE在linux环境运行

 参考:OpenKE 的探索使用(VM 的巨坑)-CSDN博客

2.配置深度学习环境(python = 3.8.0 + pytorch = 1.11.0 + cuda = 12.3)

参考:https://angxiao.blog.csdn.net/article/details/122850626

3.复现TransE

遇到问题:4060显卡的架构是sm_89,pytorch1.11.0不支持这么高的架构,能运行,但是只能用cpu,训练速度很慢

解决:额,发现是因为忘记切换虚拟环境了……之前配置深度学习环境的时候的时候遇到了这个问题,当时是因为直接下载了最新版的pytorch,那个版本还没有完善sm_89架构,解决办法是下载更加古早一点的版本,要注意兼顾python和cuda的版本。

进入之前配置好的虚拟环境openke之后就没有这个问题了

训练结果不如论文的好

论文结果

复现结果

MR(raw)

243

368

MRfilter

125

228

hit@10raw

34.9%

32.0%

hit@10filter

47.1%

47.7%

4.复现ComplEx

因为项目需要对一对多的三元组进行评分,所以拟使用ComplEx作为知识嵌入模型进行训练。

结果堪称炸裂

论文结果

复现结果

MRR(raw)

0.587

0.276

MRR(filter)

0.941

0.394

hit@1

0.936

0.343

hit@3

0.945

0.424

hit@10

0.947

0.474

  • 14
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值