Ich versuche, mehr LSTM Schichten zu meinem neuronales Netz hinzuzufügen, aber ich erhalte die folgende Fehlermeldung erhalten:Ausgabe versucht, gestapelt LSTM Schichten mit Keras zu implementieren
ValueError: Error when checking target: expected dense_4 to have 2 dimensions, but got array with shape (385, 128, 1)
Der Code für mein Modell ist wie folgt:
model = Sequential()
model.add(LSTM(60, return_sequences=True, input_shape=(128, 14)))
model.add(LSTM(60, return_sequences=False))
model.add(Dense(1))
model.compile(loss='mean_squared_error', optimizer='adam')
model.fit(data_train, RUL_train, epochs=number_epochs, batch_size=batch_size, verbose=1)
Es funktioniert gut, wenn ich die zweite LSTM-Schicht entferne. Oder wenn ich dichtere Schichten hinzufüge. Nur nicht, wenn ich die LSTM-Ebene hinzufüge. RUL_train hat Form (385, 128, 1). Der Ausgang des model.summary ist wie folgt:
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
lstm_15 (LSTM) (None, 128, 60) 18000
_________________________________________________________________
lstm_16 (LSTM) (None, 60) 29040
_________________________________________________________________
dense_7 (Dense) (None, 1) 61
=================================================================
Total params: 47,101
Trainable params: 47,101
Non-trainable params: 0
_________________________________________________________________
Jede Hilfe sehr geschätzt.
Welche Version von Keras verwenden Sie? – Brendan
'return_sequences = True' auf der letzten Ebene – DJK