Keras中的多个输出

Keras中的多个输出,第1张

Keras中的多个输出
from keras.models import Modelfrom keras.layers import *#inp is a "tensor", that can be passed when calling other layers to produce an output inp = Input((10,)) #supposing you have ten numeric values as input#here, SomeLayer() is defining a layer, #and calling it with (inp) produces the output tensor xx = SomeLayer(blablabla)(inp) x = SomeOtherLayer(blablabla)(x) #here, I just replace x, because this intermediate output is not interesting to keep#here, I want to keep the two different outputs for defining the model#notice that both left and right are called with the same input x, creating a forkout1 = LeftSideLastLayer(balbalba)(x)    out2 = RightSideLastLayer(banblabala)(x)#here, you define which path you will follow in the graph you've drawn with layers#notice the two outputs passed in a list, telling the model I want it to have two outputs.model = Model(inp, [out1,out2])model.compile(optimizer = ...., loss = ....) #loss can be one for both sides or a list with different loss functions for out1 and out2model.fit(inputData,[outputYLeft, outputYRight], epochs=..., batch_size=...)


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5647058.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存