Merge pull request #10 from DuanXu-97/patch-1

fix a bug in MultiKE_model.py
nju-websoft · Jun 16, 2020 · e4e3cd2 · e4e3cd2
2 parents a210a0c + e801730
commit e4e3cd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/MultiKE_model.py b/code/MultiKE_model.py
@@ -282,7 +282,7 @@ def save(self):
         rv_ent_embeds = self.rv_ent_embeds.eval(session=self.session)
         av_ent_embeds = self.av_ent_embeds.eval(session=self.session)
         rel_embeds = self.rel_embeds.eval(session=self.session)
-        att_embeds = self.rel_embeds.eval(session=self.session)
+        att_embeds = self.attr_embeds.eval(session=self.session)
         save_embeddings(self.out_folder, self.kgs, ent_embeds, nv_ent_embeds, rv_ent_embeds, av_ent_embeds,
                         rel_embeds, att_embeds)