框架推广

[mobius10]

如果我想把DIFT框架应用到其他KGC模型上，比如RotatE，请问要如何获取dataset/{FB15k-237 | WN18RR}/data_KGELlama文件集合？

[NJULY]

1. 所有实体的embedding表示
2. 对于query=(h, r, ?), 需要获取 (1) KGC模型预测的Top-K实体; (2) query的embedding表示, 以TransE为例, 目标是 h+r ≈ t, 那么query的embedding就是 h+r. 如果query是(?, r, t), 那么query的表示就是t-r.

[Maggie239]

请问你有做成功嘛？我也想将这个框架应用到KGC的框架上，自己写相关的嵌入部分。