是否可以在keras中将中间层设置为输出层

Question

我想尝试一下关于自动编码器的想法。模型是这样的：input (pictures) - conv2d - pooling - dense - dense(supervised output) - dense - conv - upsampling - output (pictures)是否有可能训练具有所需输出的 NNdense(supervised output)和output (pictures)? 换句话说，我想制作一个分类器并返回。

Answer 1

这可以通过 Keras 功能 API ( 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])