Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
haidfs authored Aug 13, 2019
1 parent 4daa8cc commit ff5530e
Showing 1 changed file with 20 additions and 20 deletions.
40 changes: 20 additions & 20 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,31 +1,31 @@
该工程代码主要是实现自己阅读过的和知识图谱相关的经典算法的代码:
1.TransE是知识图谱中知识表示的经典算法,工程实现了训练代码(多进程通信版)和测试代码
后续如继续进行论文阅读会补充相应的代码
2.由于data文件过大,无法上传,请至https://github.com/thunlp/KB2E下载data.zip并解压至工程的data路径
3.TransE论文地址: https://www.utc.fr/~bordesan/dokuwiki/_media/en/transe_nips13.pdf
###训练部分
####Simple版本
该工程代码主要是实现自己阅读过的和知识图谱相关的经典算法的代码:
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

### 训练部分
#### Simple版本
./train_fb15k.sh 0
仅仅使用Python完成对应的训练代码
####Manager版本
#### Manager版本
./train_fb15k.sh 1
将TransE类的实例在多进程之间传递
####Queue版本
#### Queue版本
./train_fb15k.sh 2
将TransE类的训练数据传入队列,减小进程开销,加快训练速度
当训练完成之后,再进行测试
###测试部分
####TestTransEMqQueue
### 测试部分
#### TestTransEMqQueue
python TestTransEMpQueue.py
多进程队列测试加速,效果不明显,单个测试例0.5s,测试结束需要近5h。
####TestMainTF
#### TestMainTF
python TestMainTF.py
tf与多进程测试加速,效果显著,测试结束仅需要8min左右。
###最终测试结果
FB15k
epochs:2000 MeanRank Hits@10
raw filter raw filter
head 320.743 192.152 29.7 41.2
tail 236.984 153.431 36.1 46.2
average 278.863 172.792 32.9 43.7
paper 243 125 34.9 47.1
### 最终测试结果

### THANKS TO
感谢两位前辈的代码,基本是在他们的基础上学习整理
https://github.com/wuxiyu/transE
https://github.com/ZichaoHuang/TransE

0 comments on commit ff5530e

Please sign in to comment.