在seq2seq模型中添加更多图层

时间:2018-04-13 20:52:24

标签: python machine-learning keras layer seq2seq

在fchollet给出的示例seq2seq代码中,如何向编码器和解码器添加更多LSTM层?我对形状有些麻烦,总的来说有点困惑。感谢。

1 个答案:

答案 0 :(得分:3)

Keras的功能性api可让您调用图层。这使您可以通过调用现有图层的输出来链接另一个图层。例如:

encoder_inputs = Input(shape=(None, num_encoder_tokens))
encoder = LSTM(latent_dim, return_sequences=True)
encoder_outputs, state_h, state_c = LSTM(latent_dim, return_state=True)(encoder(encoder_inputs))