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')
