Я не могу понять основную структуру модели LSTM.
Вот моя модель:
def build_model(train,n_input):
train_x, train_y = to_supervised(train, n_input)
verbose, epochs, batch_size = 1, 60,20
n_timesteps, n_features, n_outputs = train_x.shape[1], train_x.shape[2], train_y.shape[1]
train_y = train_y.reshape((train_y.shape[0], train_y.shape[1], 1))
model = Sequential()
model.add(LSTM(200, activation='relu', input_shape=(n_timesteps, n_features)))
model.add(RepeatVector(n_outputs))
model.add(LSTM(200, activation='relu', return_sequences=True))
model.add(TimeDistributed(Dense(100, activation='relu')))
model.add(TimeDistributed(Dense(1)))
model.compile(loss='mse', optimizer='adam', metrics=['accuracy'])
model.fit(train_x, train_y, epochs=epochs, batch_size=batch_size, verbose=verbose)
return model
Вот my model.summary()
Layer (type) Output Shape Param #
=================================================================
lstm_5 (LSTM) (None, 200) 172000
_________________________________________________________________
repeat_vector_3 (RepeatVecto (None, 7, 200) 0
_________________________________________________________________
lstm_6 (LSTM) (None, 7, 200) 320800
_________________________________________________________________
time_distributed_5 (TimeDist (None, 7, 100) 20100
_________________________________________________________________
time_distributed_6 (TimeDist (None, 7, 1) 101
=================================================================
Total params: 513,001
Trainable params: 513,001
Non-trainable params: 0
_________________________________________________________________
None
Из приведенного выше резюме я не понимаю, что такое lstm_5 или lstm_6. Также не указывается номер скрытого слоя в сети
Пожалуйста, кто-нибудь, помогите мне понять, сколько в этой модели скрытого слоя с нейроном.
Я в основном путаю с add(LSTM(200 ...)
и add(TimeDistributed(Dense(100..)
Я думаю, что 200 и 100 - это число нейронов в скрытом слое, и есть 4 скрытых слоя, содержащих все .add()
.
Пожалуйста, поправьте меня и проясните мои сомнения. Если возможно, попытайтесь понять по диаграмме.