Python用pytorch实现了几种知识图谱表示算法
该项目整理了知识图谱表示常用的四个数据集,提供了数据清洗整理的代码,用PyTorch实现了四种基于平移的算法。在评估阶段,使用了多进程加速,将MeanRank和Hits@10的评估时间压缩到小于1分钟。
文件列表
Python-用pytorch实现了几种知识图谱表示算法.zip
(预估有个75文件)
knowledge_representation_pytorch-master
loss.py
1KB
transH_pytorch.py
15KB
transD_pytorch.py
17KB
utils.py
5KB
transR_Bernoulli_pytorch.py
17KB
__pycache__
loss.cpython-36.pyc
2KB
utils.cpython-36.pyc
5KB
evaluation.py
28KB
暂无评论