def initialize_tbcnn_weights(clz):
clz.initialize_embedding_weights()
# Don't train We
tf.get_collection_ref(tf.GraphKeys.TRAINABLE_VARIABLES).remove(clz.get('We'))
clz.create_variable('Wcomb1', (hyper.word_dim, hyper.word_dim),
tf.constant_initializer(-.2, .2))
clz.create_variable('Wcomb2', (hyper.word_dim, hyper.word_dim),
tf.random_uniform_initializer(-.2, .2))
clz.create_variable('Wconvt', (hyper.word_dim, hyper.conv_dim),
tf.random_uniform_initializer(-.2, .2))
clz.create_variable('Wconvl', (hyper.word_dim, hyper.conv_dim),
tf.random_uniform_initializer(-.2, .2))
clz.create_variable('Wconvr', (hyper.word_dim, hyper.conv_dim),
tf.random_uniform_initializer(-.2, .2))
clz.create_variable('Bconv', (hyper.conv_dim,),
tf.random_uniform_initializer(-.2, .2))
clz.create_variable('FC1/weight', (hyper.conv_dim, hyper.fc_dim),
tf.random_uniform_initializer(-.2, .2))
clz.create_variable('FC1/bias', (hyper.fc_dim,),
tf.random_uniform_initializer(-.2, .2))
clz.create_variable('FC2/weight', (hyper.fc_dim, hyper.output_dim),
tf.random_uniform_initializer(-.2, .2))
clz.create_variable('FC2/bias', (hyper.output_dim, ),
tf.random_uniform_initializer(-.2, .2))
评论列表
文章目录