def save(self):
ent_embeds = self.ent_embeds.eval(session=self.session)
rel_embeds = self.rel_embeds.eval(session=self.session)
mapping_mat_1 = self.mapping_mat_1.eval(session=self.session)
mapping_mat_2 = self.mapping_mat_2.eval(session=self.session)
rd.save_embeddings(self.out_folder, self.kgs, ent_embeds, rel_embeds, None,
mapping_mat=mapping_mat_1, rev_mapping_mat=mapping_mat_2)
def save(self):
embedding = self.sess.run(self.output)
rd.save_embeddings(self.out_folder,
self.kgs,
embedding,
None,
None,
mapping_mat=None)
def save(self):
ent_embeds = self.vec_se
attr_embeds = self.vec_ae
rd.save_embeddings(self.out_folder,
self.kgs,
ent_embeds,
None,
attr_embeds,
mapping_mat=None)
def save(self):
embeds_list = list()
input_embeds = self.init_embedding
output_embeds_list = self.output_embeds_list
for output_embeds in [input_embeds] + output_embeds_list:
output_embeds = tf.nn.l2_normalize(output_embeds, 1)
output_embeds = np.array(output_embeds.eval(session=self.session))
embeds_list.append(output_embeds)
ent_embeds = np.concatenate(embeds_list, axis=1)
rd.save_embeddings(self.out_folder, self.kgs, ent_embeds, None, None, mapping_mat=None)
def save(self):
ent_embeds = self.ent_embeds.eval(session=self.session)
rel_embeds = self.rel_embeds.eval(session=self.session)
mapping_mat = self.mapping_mat.eval(
session=self.session) if self.mapping_mat is not None else None
rd.save_embeddings(self.out_folder,
self.kgs,
ent_embeds,
rel_embeds,
None,
mapping_mat=mapping_mat)
def save(self):
ent_embeds = self.vec_se
if self.args.test_method == "sa":
attr_embeds = self.vec_ae
else:
attr_embeds = None
rd.save_embeddings(self.out_folder,
self.kgs,
ent_embeds,
None,
attr_embeds,
mapping_mat=None)
def save(self):
ent_embeds = self.re_ent_embeds.eval(
session=self.session) + self.im_ent_embeds.eval(
session=self.session)
ent_embeds = preprocessing.normalize(ent_embeds)
rel_embeds = self.rel_embeds.eval(session=self.session)
mapping_mat = self.mapping_mat.eval(
session=self.session) if self.mapping_mat is not None else None
rd.save_embeddings(self.out_folder,
self.kgs,
ent_embeds,
rel_embeds,
None,
mapping_mat=mapping_mat)
def save(self):
embeds = self.embeds.eval(session=self.session)
read.save_embeddings(self.out_folder, self.kgs, None, None, embeds)
