python - 如何在Keras中的自定义损失函数中计算True Positive

我用Keras做分类任务，我在Keras中制作了简单的自定义损失函数，并且可以正常工作

import keras.backend as K

def customLoss(yTrue,yPred):

    return K.abs(yTrue-yPred)

为了使我想要更复杂的损失函数，我需要计算真正，真负，假正，假负

如何计算？

我无法计算它们，因为我不知道yTrue和yPred的类型。他们是2D数组或列表还是其他东西。如果我知道，也许我可以使用for来计算TP，TN，FP，FN，如下所示：

TP=0
for x,y in zip(yTrue,yPred):
   if x == 1 and y > 0.5:
      TP=TP+1

最佳答案

根据Keras Documentation，yTrue / yPred的数据类型为TensorFlow / Theano张量，具体取决于所使用的后端。

因此，您不能对损失函数使用for循环，否则会出现错误。

但是，您可以使用逻辑的方法进行处理：

TN = np.logical_and(K.eval(y_true) == 0, K.eval(y_pred) == 0)
FP = np.logical_and(K.eval(y_true) == 0, K.eval(y_pred) == 1)

之后，您可以将它们加起来：

TN = K.sum(K.variable(TN))
FP = K.sum(K.variable(FP))