LunaBlack / KGAT-pytorch

324 stars 73 forks source link

Question about pretrain embedding #3

Open WDdeBWT opened 4 years ago

WDdeBWT commented 4 years ago

您好,我想问一下代码里面用到的pretrain embedding是使用TransR训练多轮得到的吗?

WDdeBWT commented 4 years ago

我在代码里没有看到关于训练并保存pretrain embedding的代码,想问一下这一块的实现,如果能够有配套的代码就更好了。感谢

LunaBlack commented 4 years ago

我用的是paper作者在他的github repo中提供的pretrain embedding:https://github.com/xiangwang1223/knowledge_graph_attention_network

我也没有看到他这部分相关的code,建议你在那个repo下提个issue问问。

PS:不用pretrain的效果貌似差很多。

WDdeBWT commented 4 years ago

我用的是paper作者在他的github repo中提供的pretrain embedding:https://github.com/xiangwang1223/knowledge_graph_attention_network

我也没有看到他这部分相关的code,建议你在那个repo下提个issue问问。

PS:不用pretrain的效果貌似差很多。

好的感谢。看您这部分代码写的非常详细,我一只以为是作者团队放出的Pytorch版本。祝您工作&研究顺利

Genius-pig commented 8 months ago

作者好像回复了,但是我依然没有找到在哪。 https://github.com/xiangwang1223/knowledge_graph_attention_network/issues/28 https://github.com/xiangwang1223/knowledge_graph_attention_network/issues/29 https://github.com/xiangwang1223/knowledge_graph_attention_network/issues/20

Keep0828 commented 8 months ago

我看原作者的解释是: In terms of the pretrain step, you can run the BPRMF model and store the MF embeddings under the 'Model/pretrain/'.

Hope it can be helpful. https://github.com/xiangwang1223/knowledge_graph_attention_network/issues/28