如何为keras(深度学习库)编写分类精度损失函数?

时间:2016-06-19 13:59:48

标签: keras

如何为keras(深度学习库)编写分类精度损失函数?

分类准确性损失是错误预测的百分比,即#错误/#数据点。

是否可以为此编写自定义丢失函数?

感谢。

1 个答案:

答案 0 :(得分:1)

修改

尽管Keras允许您使用自定义丢失功能,但我不相信使用精确度会因为丢失有意义。首先,网络的最后一层通常是soft-max,因此您可以获得类概率的向量,而不是单个最可能的类。其次,我担心由于缺乏准确性而导致梯度计算存在问题。

旧帖子

Keras为您提供使用自定义丢失功能的可能性。要获得准确性损失,您可以从已经实施的examples中获取灵感。对于二进制分类,我建议以下实现

def mean_accuracy_error(y_true, y_pred):
    return K.mean(K.abs(K.sign(y_true - y_pred)), axis=-1)