ZJU-Fangyin / KCL

Code and Data for the paper: Molecular Contrastive Learning with Chemical Element Knowledge Graph [AAAI 2022]
MIT License
86 stars 11 forks source link

关于使用知识图谱嵌入的问题 #12

Closed 66246764 closed 1 year ago

66246764 commented 1 year ago

作者您好,我又来请教问题了- - 您在readme中Knowledge feature initialization部分提到如果想要自己训练,可以运行code/initial 的load.py,但是在该文件中是通过36行代码: relation_emb, entity_emb = load('RotatE_128_64.pkl', total_entity2id=data.entity2id, total_relation2id=data.relation2id, e_dim=128, r_dim=64) 实现的,而这里的'RotatE_128_64.pkl'文件本身就包含了您使用化学周期表初始化的嵌入值,如下图 image

请问您能给出获得'RotatE_128_64.pkl'这个文件的例子吗,entity和relation的值是怎么得到的呢?