def _define_variables(self): with tf.variable_scope('relational' + 'embeddings'): self.ent_embeds = init_embeddings( [self.kgs.entities_num, self.args.dim], 'ent_embeds', self.args.init, self.args.ent_l2_norm, dtype=tf.float32) self.rel_embeds = init_embeddings( [self.kgs.relations_num, self.args.dim], 'rel_embeds', self.args.init, self.args.rel_l2_norm, dtype=tf.float32) with tf.variable_scope('character' + 'embeddings'): self.ent_embeds_ce = init_embeddings( [self.kgs.entities_num, self.args.dim], 'ent_embeds_ce', self.args.init, self.args.ent_l2_norm, dtype=tf.float32) self.attr_embeds = init_embeddings( [self.kgs.attributes_num, self.args.dim], 'attr_embeds', self.args.init, self.args.attr_l2_norm, dtype=tf.float32) self.char_embeds = init_embeddings( [self.char_list_size, self.args.dim], 'char_embeds', self.args.init, self.args.char_l2_norm, dtype=tf.float32) self.value_id_char_ids = tf.constant(self.value_id_char_ids)
def _define_variables(self): with tf.variable_scope('relation_view' + 'embeddings'): self.rv_ent_embeds = init_embeddings([self.kgs.entities_num, self.args.dim], 'rv_ent_embeds', self.args.init, self.args.ent_l2_norm) self.rel_embeds = init_embeddings([self.kgs.relations_num, self.args.dim], 'rel_embeds', self.args.init, self.args.rel_l2_norm) with tf.variable_scope('image_view' + 'embeddings'): self.iv_ent_embeds = init_embeddings([self.kgs.entities_num, self.args.dim], 'iv_ent_embeds', self.args.init, self.args.ent_l2_norm) self.iv_ent_mapping = init_embeddings([self.args.vgg_dim, self.args.dim], 'iv_ent_mapping', self.args.init, self.args.ent_l2_norm) self.image_embeds = tf.constant(self.img_embeds, dtype=tf.float32) with tf.variable_scope('attribute_view' + 'embeddings'): self.av_ent_embeds = xavier_init([self.kgs.entities_num, self.args.dim], 'av_ent_embeds', True) self.av_c = xavier_init([self.kgs.attributes_num, self.args.rbf_dim], 'av_c', True) self.av_delta = xavier_init([self.kgs.attributes_num, self.args.rbf_dim], 'av_delta', True) self.av_W = xavier_init([self.args.rbf_dim, self.args.dim], 'av_W', True) self.av_b = xavier_init([1, self.args.dim], 'av_b', True) # False important! self.attr_embeds = xavier_init([self.kgs.attributes_num, self.args.dim], 'attr_embeds', False) with tf.variable_scope('shared' + 'embeddings'): self.ent_embeds = init_embeddings([self.kgs.entities_num, self.args.dim], 'ent_embeds', self.args.init, self.args.ent_l2_norm) with tf.variable_scope('shared' + 'combination'): self.rv_mapping = tf.get_variable('rv_mapping', shape=[self.args.dim, self.args.dim], initializer=tf.initializers.orthogonal()) self.iv_mapping = tf.get_variable('iv_mapping', shape=[self.args.dim, self.args.dim], initializer=tf.initializers.orthogonal()) self.eye_mat = tf.constant(np.eye(self.args.dim), dtype=tf.float32, name='eye')
def _define_variables(self): with tf.variable_scope('relational' + 'embeddings'): # initialize embeddings for entity and relations (BootEA -> normal distribution l2 normalized) self.ent_embeds = init_embeddings([self.kgs.entities_num, self.args.dim], 'ent_embeds', self.args.init, self.args.ent_l2_norm) self.rel_embeds = init_embeddings([self.kgs.relations_num, self.args.dim], 'rel_embeds', self.args.init, self.args.rel_l2_norm)
def _define_variables(self): with tf.variable_scope('relational' + 'embeddings'): self.ent_embeds = init_embeddings( [self.kgs.entities_num, self.args.dim], 'ent_embeds', self.args.init, self.args.ent_l2_norm) self.rel_embeds = init_embeddings( [self.kgs.relations_num, self.args.dim], 'rel_embeds', self.args.init, self.args.rel_l2_norm)
def _define_variables(self): with tf.variable_scope('relational' + 'embeddings'): self.ent_embeds = init_embeddings([self.kgs.entities_num, self.args.dim], 'ent_embeds', self.args.init, self.args.ent_l2_norm) self.rel_embeds = init_embeddings([self.kgs.relations_num, self.args.dim], 'rel_embeds', self.args.init, self.args.rel_l2_norm) with tf.variable_scope('prob' + 'parameters'): self.entity_w = init_embeddings([self.kgs.entities_num, self.args.dim], 'entity_w', 'xavier', False) self.entity_b = init_embeddings([self.kgs.entities_num, ], 'entity_b', 'xavier', False)
def _define_variables(self): with tf.variable_scope('relational' + 'embeddings'): self.ent_embeds = init_embeddings([self.kgs.entities_num, self.args.dim], 'ent_embeds', self.args.init, self.args.ent_l2_norm) self.rel_embeds = init_embeddings([self.kgs.relations_num, self.args.dim], 'rel_embeds', self.args.init, self.args.rel_l2_norm) with tf.variable_scope('kgs' + 'mapping'): self.mapping_mat_1 = orthogonal_init([self.args.dim, self.args.dim], 'mapping_matrix_1') self.eye_mat_1 = tf.constant(np.eye(self.args.dim), dtype=tf.float32, name='eye_1') self.mapping_mat_2 = orthogonal_init([self.args.dim, self.args.dim], 'mapping_matrix_2') self.eye_mat_2 = tf.constant(np.eye(self.args.dim), dtype=tf.float32, name='eye_2')