Размер поезда: 13974 Размер теста: 3494 Форма поезда: (13974, 101, 6, 5) Форма теста: (3494, 101, 6, 5) Форма входа поезда: (13974, 100, 6, 5) Выход поездаshape: (13974, 1, 6, 5) тестовая входная форма: (3494, 100, 6, 5) тестовая выходная форма: (3494, 1, 6, 5)
model = Sequential()
model.add(TimeDistributed(Conv2D(32, (3, 3), padding='same'),
input_shape=(100, 6, 5,1)))
model.add(TimeDistributed(Activation('relu')))
model.add(TimeDistributed(Conv2D(32, (3, 3))))
model.add(TimeDistributed(Activation('relu')))
model.add(TimeDistributed(MaxPooling2D(pool_size=(2, 2))))
model.add(TimeDistributed(Dropout(0.25)))
model.add(TimeDistributed(Flatten()))
model.add(TimeDistributed(Dense(512)))
model.add(TimeDistributed(Dense(35, name="first_dense_flow" )))
model.add(LSTM(20, return_sequences=True, name="lstm_layer_flow"));
model.add(TimeDistributed(Dense(101), name=" time_distr_dense_one_ flow "))
model.add(GlobalAveragePooling1D(name="global_avg_flow"))
model.compile(loss='mae', optimizer='adam', metrics=['accuracy'])
model.fit(train_input,train_output,epochs=50,batch_size=60)
Я получил ValueError:Ошибка при проверке ввода: ожидалось, что time_distributed_38_input будет иметь 5 измерений, но получил массив с формой (13974, 100, 6, 5)
Мне нужно предсказать (1,6,5) из (100,6,5)) где 100 - отметка времени.
Пожалуйста, исправьте любые изменения, необходимые в модели