栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

Keras中的多个输出

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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=...)


转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/641269.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号