JiapengWu / TeMP

Temporal Message Passing Network for Temporal Knowledge Graph Completion
55 stars 18 forks source link

impute使用问题 #3

Closed chuhang123 closed 3 years ago

chuhang123 commented 3 years ago

你好,首先非常感谢您提供代码。在复现代码的过程中,发现只有GRRGCN模型使用了impute创新点,BiGRRGCN、SARGCN、BiSARGCN并没有使用impute,是这样么?频率门也是这样的 我debug模型时只有GRRGCN模型停留在断点处了(在RRGCN.py的276行,使用impute的地方),而另外三个模型没有。是我没有找到正确的代码位置么?希望大佬指教,非常感谢

NeuSyz commented 3 years ago

我在训练BiSARGCN、SRGCN时候会卡在某一轮

JiapengWu commented 3 years ago

Hello,

不好意思回的有点晚。是的 impute没有用在self-attention model里面,原因在Appendix A.2: Intrinsic imputation for TeMP-SA. (https://arxiv.org/pdf/2010.03526.pdf). Frequency-basd gating是可以用的,直接在command里加上--post-aggregation就行了: [cid:DBEAE30C-2C24-4A98-9FD8-25D81913C7F7@Home] GRRGCN和BiGRRGCN都用了imputation。BiGRRGCN用到的地方在models/BiRRGCM.py的79, 311-312和329-330.

另外。。其实ablation中imputation没有很明显的作用,只用vanilla model + FG其实效果就不错。

On Apr 7, 2021, at 8:24 AM, 楚航 @.**@.>> wrote:

你好,首先非常感谢您提供代码。在复现代码的过程中,发现只有GRRGCN模型使用了impute创新点,BiGRRGCN、SARGCN、BiSARGCN并没有使用impute,是这样么?频率门也是这样的 我debug模型时只有GRRGCN模型停留在断点处了(在RRGCN.py的276行,使用impute的地方),而另外三个模型没有。是我没有找到正确的代码位置么?希望大佬指教,非常感谢

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHubhttps://github.com/JiapengWu/TeMP/issues/3, or unsubscribehttps://github.com/notifications/unsubscribe-auth/AFN2LV4YVQD7OXURH2PEIM3THRFJPANCNFSM42QWAE7A.

JiapengWu commented 3 years ago

Hi, I have just uploaded some trained models, please check out the updated readme.