如何向合并的keras模型添加层?

2024-04-28 06:17:13 发布

您现在位置:Python中文网/ 问答频道 /正文

我想向合并模型(在输出上)添加两个层,一个是relu层,一个是softmax层,这里是我的合并模型:

concat = concatenate([model1.output, model2.output], name='Concatenate')
final_model_output = Dense(20, activation='softmax')(concat)
final_model = Model(inputs=[model1.input, model2.input], outputs=final_model_output,
                name='Final_output')

final_model.compile(loss='sparse_categorical_crossentropy',
              optimizer='adam',
              metrics=['accuracy'])

1条回答
网友
1楼 · 发布于 2024-04-28 06:17:13

我想答案已经在你所附的代码中了

您可以像在第二行中那样添加更多层

# model_name = Layer_name(...)(model_name)
final_model_output = Dense(20, activation='softmax')(concat)

例如:

concat = concatenate([model1.output, model2.output], name='Concatenate')
x = Dense(20, activation='softmax')(concat)
x = Dense(20, activation='relu')(x)
final_model_output = Dense(20, activation='softmax')(x)
final_model = Model(inputs=[model1.input, model2.input], outputs=final_model_output,
                name='Final_output')

相关问题 更多 >