model.py 文件源码

def __init__(self, vocab_size, learning_rate=LEARNING_RATE, grad_clip=GRAD_CLIP, init_embedding=Normal()):
        self.vocab_size = vocab_size
        self.lr = learning_rate
        self.gc = grad_clip
        self.W = init_embedding
        if USE_GRU:
            self.rnn_layer = GRULayer
        else:
            self.rnn_layer = LSTMLayer

        if CONSTANTLY_FEED_HIDDEN_STATE:
            self.net = self._get_feed_net() # seq2seq v2
        else:
            self.net = self._get_net() # seq2seq v1
        self.train = self._get_train_fun()
        self.predict = self._get_predict_fun()
        # self.encode = self._get_encoder_fun()
        # self.decode = self._get_decoder_fun()
        # self.embedding = self._get_embedding_fun()
        # self.slicing = self._get_slice_fun()
        # self.decoding = self._get_dec_fun()