ZichaoHuang / TransE

A TensorFlow implementation of TransE model
198 stars 62 forks source link

embedding结果与entity、relation是如何对应的 #10

Closed Wangwxy closed 6 years ago

Wangwxy commented 6 years ago

您好。请问,我把得到的embedding打印出来,这些embedding与entity、relation的对应关系是怎样的?

ZichaoHuang commented 6 years ago

entity2id.txtrelation2id.txt两个文件分别是实体和关系所对应的index,也就是embedding中对应的index。

Wangwxy commented 6 years ago

谢谢您。您的意思,就entity来说,是打印出的embedding的顺序就是entity2id.txt中的entity的顺序吗? 我现在只能打印出来embedding,不能把相应的id打印出来。所以有这个疑问。

ZichaoHuang commented 6 years ago

是的。 entity2id.txtrelation2id.txt实际上只是从0开始的index映射,而self.entity_embedding实际上可以看作是一个n*k的矩阵,n是实体的个数,k是实体embedding的维度,self.entity_embedding的第0行对应的就是index为0的实体的embedding。 self.relation_embedding也同理。

Wangwxy commented 6 years ago

谢谢!