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)