如何在LSTM中修复'batch_size'

时间:2019-04-05 02:04:09

标签: python lstm

我正在运行LSTM代码,并且我想使其成为双向LSTM。我该怎么办?

我正在使用https://github.com/brunnergino/JamBot.git中的代码。名为 polyphonic_lstm_training.py 的笔记本具有代码。

model = Sequential()
model.add(LSTM(lstm_size, batch_size=batch_size, input_shape=(step_size, new_num_notes+chord_dim+counter_size), stateful=True))
model.add(LSTM(lstm_size, batch_input_shape=(batch_size,step_size, new_num_notes+chord_dim+counter_size), stateful=True))

我希望它可以使用双向LSTM进行训练

1 个答案:

答案 0 :(得分:0)

实现双向LSTM的一种可能方法是在第二个LSTM层处理输入之前反转输入。然后再次反转第二个LSTM的输出,并与第一个LSTM层的输出连接。