请和我一起对我很陌生。
我正在使用LSTM训练分类器,并具有以下代码
我在第三个LSTM层说尺寸存在问题时遇到问题
我的训练集的形状为(34799, 32, 32)
model = Sequential()
model.add(LSTM(64, activation = 'relu', input_shape=X_train[0].shape, return_sequences=True))
model.add(Dropout(0.25))
model.add(LSTM(128, activation = 'relu'))
model.add(Dropout(0.25))
model.add(LSTM(128, activation = 'relu'))
model.add(Dropout(0.25))
model.add(LSTM(64, activation = 'relu'))
model.add(Dropout(0.25))
model.add(Dense(len(sign_names), activation='softmax'))
每个LSTM层的返回序列都可以解决问题。