python - 如何使用张量实现平衡错误率？

Question

我正在尝试使用 Keras 解决二进制分类问题，并且必须满足给定的平衡错误率。我对 BER 函数的实现是：

def balanced_error_rate(y_true, y_pred):
    TN, FP, FN, TP = confusion_matrix(y_true, y_pred).ravel()
    return 0.5 * (FP/(TN + FP) + FN/(FN + TP))

但是，我意识到我需要 BER 为张量格式。我知道我的问题类似于：Balanced Error Rate as metric function。但是，我无法判断该实现有什么问题。如果有人对如何使用张量实现 BER 有任何建议，我将不胜感激！

标签： pythontensorflowkeras