小编Ahm*_*ael的帖子

在 Keras 中为模型编译指定多个损失函数

我想为交叉熵的对象类指定 2 个损失函数,另一个为均方误差的边界框。如何在 model.compile 每个输出中指定相应的损失函数?

model = Sequential()

model.add(Dense(128, activation='relu'))
out_last_dense = model.add(Dense(128, activation='relu'))
object_type = model.add(Dense(1, activation='softmax'))(out_last_dense)
object_coordinates = model.add(Dense(4, activation='softmax'))(out_last_dense)

/// here is the problem i want to specify loss function for object type and coordinates
model.compile(loss= keras.losses.categorical_crossentropy,
   optimizer= 'sgd', metrics=['accuracy'])
Run Code Online (Sandbox Code Playgroud)

python deep-learning keras tensorflow

3
推荐指数
1
解决办法
3173
查看次数

标签 统计

deep-learning ×1

keras ×1

python ×1

tensorflow ×1