У меня есть проблема, которая связана с предсказанием четырех выходных данных, когда задан вектор предикторов. Выдает ошибку на входе слоя LSTM.
У меня есть
X.shape,Y.shape = ((2300, 36, 768), (2300, 4, 54))
# Core Part
checkpoint = ModelCheckpoint('model-{epoch:03d}-{acc:03f}-{val_acc:03f}.h5', verbose=1, monitor='val_loss',save_best_only=True, mode='auto')
data_dim = 768
timesteps = X.shape[1]
num_classes = 10
# # expected input data shape: (batch_size, timesteps, data_dim)
model = Sequential()
model.add(LSTM(256, return_sequences=True, input_shape=(timesteps, data_dim)))
model.add(LSTM(64, return_sequences=True))
model.add(LSTM(32))
model.add(Dense(54, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer='rmsprop', metrics=['accuracy'])
model.fit(X, Y, batch_size=64, epochs=10, validation_split=0.1)
Приведенный выше код работает нормально, когда у меня есть X.shape,Y.shape = ((2300, 36, 768), (2300, 15))
Как мне преодолеть Исходя из этого, и как я могу установить уровни LSTM и DENSE, если у меня более одного, четырех или десяти выводов прогноза?
Заранее спасибо.