小编ter*_*low的帖子

Keras 中的 Seq2Seq 双向编码器解码器

我正在尝试使用 Keras 实现一个 seq2seq 编码器-解码器,在编码器上使用双向 lstm,如下所示:

from keras.layers import LSTM,Bidirectional,Input,Concatenate
from keras.models import Model

n_units = 8
n_input = 1
n_output = 1

# encoder
encoder_inputs = Input(shape=(None, n_input))
encoder = Bidirectional(LSTM(n_units, return_state=True))
encoder_outputs, forward_h, forward_c, backward_h, backward_c = encoder(encoder_inputs)
state_h = Concatenate()([forward_h, backward_h])
state_c = Concatenate()([forward_c, backward_c])
encoder_states = [state_h, state_c]

# decoder
decoder_inputs = Input(shape=(None, n_output))    
decoder_lstm = LSTM(n_units*2, return_sequences=True, return_state=True)
decoder_outputs, _, _ = decoder_lstm(decoder_inputs, initial_state=encoder_states)
Run Code Online (Sandbox Code Playgroud)

这是我在最后一行遇到的以下错误:

ValueError: Dimensions must be equal, but are 8 and 16 …
Run Code Online (Sandbox Code Playgroud)

neural-network lstm keras rnn seq2seq

3
推荐指数
1
解决办法
6574
查看次数

标签 统计

keras ×1

lstm ×1

neural-network ×1

rnn ×1

seq2seq ×1