MaximTian / TransX

Trans系列之TransE and TransH, PTransE
33 stars 12 forks source link

TransX

知识表示Trans系列之TransE、TransH、TransR以及PTransE算法的JAVA实现

代码主要参考清华NLP实验室的代码内容

代码的原理来源于几篇TransE相关的论文

  1. TransE
  2. TransH
  3. TransR
  4. PTransE

运行环境

jdk-1.8.0

代码结构

格式说明

entity2id.txt中,输入格式为:

{实体名称} \t {id}

relation2id.txt中,输入格式为:

{关系名称} \t {id}

train.txt\test.txt 格式为:

{head} \t {tail} \t {relaiton}