如何在Keras中实现加权交叉熵损失?

时间:2019-03-18 03:41:05

标签: keras loss-function

我想知道如何在Keras的二进制或多类分类器中为损失函数添加自定义权重。我正在使用binary_crossentropy或sparse_categorical_crossentropy作为基线,我希望能够选择权重来为每个类给出错误的预测。

1 个答案:

答案 0 :(得分:1)

对于多个类,不应使用二进制,而应使用分类交叉熵。

考虑使用自定义损失功能,如下所述:Custom loss function in Keras