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

问题描述 投票:1回答:1

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

keras loss-function
1个回答
1
投票

对于多个类,应该使用非二元但是分类的交叉熵。

考虑使用如下所述的自定义丢失函数:Custom loss function in Keras

© www.soinside.com 2019 - 2024. All rights reserved.