zqhead / transe Goto Github PK

你好我想问下，如果我想用这个模型训练FB5M数据集，想得到entity.bin和relation.bin文件怎么做？

你好！
关于在TransE_pytorch中的 hpt 和 tph， 文件中的注释是这样说的:"这里关于p的说明 tph 表示每一个头结对应的平均尾节点数 hpt 表示每一个尾节点对应的平均头结点数"。而关于hpt和tph的代码是这一段`
for r_ in relation_head:
sum1, sum2 = 0, 0
for head in relation_head[r_]:
sum1 += 1
sum2 += relation_head[r_][head]
tph = sum2 / sum1
relation_tph[r_] = tph

for r_ in relation_tail:
    sum1, sum2 = 0, 0
    for tail in relation_tail[r_]:
        sum1 += 1
        sum2 += relation_tail[r_][tail]
    hpt = sum2 / sum1
    relation_hpt[r_] = hpt`

以tph的代码为例，这里的sum1是关系r_对应的头结点数，而sum2似乎并非是关系r_对应的尾结点数，而应该是关系r_与头结点的连接数，当然，这与关系r_与尾结点的连接数是一样的，但是不管怎样，我认为sum2不应该是关系r_对应的尾节点数。这和下文注释中的表达是矛盾的。

FileNotFoundError: [Errno 2] No such file or directory: 'WN18_torch_TransE_entity_20dim_batch400'

At first this is a amazing code work! and after running the code, the code showed the error that the 'WN18_torch_TransE_entity_50dim_batch4800' file can not be found, and i cannot find it in the whole file of the TransE file.