seq_to_seq_autoencoder.py 文件源码

def _create_layers(self, input_layer):

        """ Create the encoding and the decoding layers of the sequence-to-sequence autoencoder.
        :return: self
        """

        encode_layer = LSTM(name='encoder',
                            units=self.n_hidden,
                            activation=self.enc_activation)(input_layer)

        n_inputs = K.int_shape(input_layer)[-1]
        decoded = RepeatVector(n=self.time_steps)(encode_layer)
        self._decode_layer = LSTM(name='decoder',
                                  units=n_inputs,
                                  activation=self.dec_activation,
                                  return_sequences=True)(decoded)