Какую дозу означает число в LSTM - PullRequest
0 голосов
/ 02 августа 2020

В нейронной сети RNN

что означает число 128 после LSTM?

# RNN Recurrent Neural Network architecture 
model = Sequential()

model.add(LSTM(128, input_shape=(X_train.shape[1:]), return_sequences=True))
#model.add(Dropout(0.2))
model.add(BatchNormalization())

Ответы [ 2 ]

2 голосов
/ 02 августа 2020

Я думаю, что следующая ссылка может дать четкий ответ на этот вопрос.

https://www.tensorflow.org/api_docs/python/tf/compat/v1/keras/layers/LSTM

Согласно ссылке, это размер выходного пространства для LSTM.

1 голос
/ 02 августа 2020

Это номер узла LSTM , который вы хотите использовать (перпендикулярно потоку информации).

...