model.py 文件源码

def __init__(self, config, mode):
        self.config = config
        self.mode = mode

        self.weight_initializer = tf.contrib.layers.xavier_initializer()
        self.const_initializer = tf.constant_initializer(0.0)
        self.emb_initializer = tf.random_uniform_initializer(minval=-1.0, maxval=1.0)

        self.level1_word2ix = json.load(open('data/train/word2ix_stem.json'))
        self.level2_word2ix = json.load(open('data/train/word2ix_attr.json'))

        self.level1_model = level1_model.Level1Model(word_to_idx=self.level1_word2ix,
                                                     dim_feature=config.LEVEL1_dim_feature,
                                                     dim_embed=config.LEVEL1_dim_embed,
                                                     dim_hidden=config.LEVEL1_dim_hidden,
                                                     alpha_c=config.LEVEL1_alpha, dropout=config.LEVEL1_dropout,
                                                     n_time_step=config.LEVEL1_T, train=(self.mode == 'training'))

        self.level2_model = level2_model.Level2Model(word_to_idx=self.level2_word2ix,
                                                     dim_feature=config.LEVEL2_dim_feature,
                                                     dim_embed=config.LEVEL2_dim_embed,
                                                     dim_hidden=config.LEVEL2_dim_hidden,
                                                     dropout=config.LEVEL2_dropout, n_time_step=config.LEVEL2_T)