python挑战赛怎么做trans_GitHub - haidfs/TransE: TransE方法的Python实现,解释SGD中TransE的向量更新...

该工程代码主要是实现自己阅读过的和知识图谱相关的经典算法的代码:

1.TransE是知识图谱中知识表示的经典算法,工程实现了训练代码(多进程通信版)和测试代码

后续如继续进行论文阅读会补充相应的代码

2.TransE论文地址: https://www.utc.fr/~bordesan/dokuwiki/_media/en/transe_nips13.pdf

3.TransE SGD解释与代码简单解释: https://blog.csdn.net/weixin_42348333/article/details/89598144

环境配置与前置技术要求:

环境配置

CPU 24 Intel(R) Xeon(R) CPU E5-2620 v3 @ 2.40GHz

内存 128G

系统 CentOS Linux release 7.5.1804 (Core)

涉及到的技术点

Linux基本操作与shell脚本简单语法 利用shell脚本去进行训练,避免使用Python文件中的main函数去训练

git基本操作

Python的多进程 可以参考https://blog.csdn.net/weixin_42348333/article/details/105126470

TensorFlow 常用API

Pycharm debug与pdb debug

对论文的训练和评估细节的深刻理解

训练部分

Simple版本

./train_fb15k.sh 0

仅仅使用Python完成对应的训练代码

Manager版本

./train_fb15k.sh 1

将TransE类的实例在多进程之间传递

Queue版本

./train_fb15k.sh 2

将TransE类的训练数据传入队列,减小进程开销,加快训练速度

注意事项

当训练完成之后,再进行测试。

测试代码需要在Linux环境执行,Windows环境多进程速度慢,且多进程有bug!!!

测试部分

TestMainTF

python TestMainTF.py

tf与多进程测试加速,效果显著,Linux环境128G服务器,测试结束仅需要8min左右。

最终测试结果

THANKS

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值