gpt4 book ai didi

python - 如何修复 LSTM 中的 'batch_size'

转载 作者:太空宇宙 更新时间:2023-11-03 21:04:18 25 4
gpt4 key购买 nike

我正在运行 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 进行训练

最佳答案

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

关于python - 如何修复 LSTM 中的 'batch_size',我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55527294/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com