Блог

составление модели

 model.compile(loss="mse", loss_weights=[0.9, 0.1], optimizer=keras.optimizers.SGD(lr=1e-3))

history = model.fit((x_train_A, x_train_B), (y_train, y_train), epochs=10,
                    validation_data=((x_valid_A, x_valid_B), (y_valid, y_valid)))

total_loss, main_loss, aux_loss = model.evaluate((x_test_A, x_test_B), (y_test, y_test))

y_pred_main, y_pred_aux = model.predict((x_new_A, x_new_B))
 

В последней строке сообщается об ошибке:

Ошибка значения: Ввод 0 слоя dense_4 несовместим со слоем: ожидаемая ось -1 входной формы должна иметь значение 6, но полученный ввод с формой (нет, 8)

вот ссылка на ячейку colab

форма x_new_B была неправильной, исправьте это с помощью:

 x_new_A, x_new_B = x_test_A[:3], x_test_B[:3]