python - 如何在 keras 中垂直堆叠训练好的模型？

Question

我在 keras 中有两个经过训练的模型，我想将一个堆叠在另一个之上以形成一个模型。我想这样做以将我训练有素的模型合二为一。

我认为 merge(Merge) 用于水平堆叠模型，而我想垂直堆叠 keras 功能 API 模型。

PS：输出 1 的形状与输入 2 相同

标签： pythontensorflowkeras

Answer 1

inputA = Input(input_shape_for_A)
outputA = modelA(inputA)
outputB = modelB(outputA)

modelC = Model(inputA, outputB)