Многослойный скрытый слой в LSTM в Керасе

x = Input(shape=(timesteps, input_dim,))

# LSTM encoding
h = LSTM(2048)(x)

Это несколько строк кода из файла, который я скачал из интернета. Я думаю h держит для однослойного слоя LSTM с 2048 единицами. Как это может сделать его многослойным, т.е. 2 скрытых слоя.

1 ответ

Решение

Просто добавьте еще один слой (давайте назовем его g)! Но так как мы переходим к другому LSTM слой, мы собираемся добавить return_sequences Параметр ключевого слова для первого слоя, чтобы мы могли получить правильную форму ввода для второго слоя.

x = Input(shape=(timesteps, input_dim,))

# LSTM encoding
h = LSTM(2048, return_sequences=true)(x)
g = LSTM(10)(h)
Другие вопросы по тегам