Можно ли установить средний слой в качестве выходного слоя в керасе?

Question

Я хотел бы попробовать идею об автоэнкодере. Модель выглядит следующим образом:

input (pictures) - conv2d - pooling - dense - dense(supervised output) - dense - conv - upsampling - output (pictures)

Можно ли обучить NN с желаемыми выходами для dense(supervised output) и output (pictures)? Другими словами, я хочу сделать классификатор и обратно.

Answer 1

Это можно сделать с помощью функционального API Keras (https://keras.io/getting-started/functional-api-guide/).

Минимальный пример, где модель имеет 2 выхода, один из промежуточного уровня и один из конечного уровня:

import keras
input = keras.layers.Input(shape=(3,))

intermediate = keras.layers.Dense(10)(input)
final_output = keras.layers.Dense(3)(intermediate)

model = keras.Model(inputs=input, outputs=[intermediate, final_output])