Пример #1
0
    def build_decoder(self):
        inputs = Input(shape=(self.hidden_dim, ))
        repeat = RepeatVector(self.max_sequence_len)(inputs)

        lstm_layer = GRU(self.hidden_dim, return_sequences=True)
        hidden_rep = lstm_layer(repeat)
        time_dist_dense_layer = TimeDistributed(
            Dense(self.vocab_size, activation='softmax'))
        outputs = time_dist_dense_layer(hidden_rep)

        lstm_layer.set_weights(
            [self.sequence_autoencoder.layers[4].get_weights()])
        time_dist_dense_layer.set_weights(
            [self.sequence_autoencoder.layers[5].get_weights()])

        decoder = Model(inputs, outputs)
        decoder.summary()
        return decoder