Keras: 评价指标 categorical_accuracy和 sparse_categorical_accuracy

本文深入探讨了Keras中用于多分类任务的两种评价函数:categorical_accuracy和sparse_categorical_accuracy。详细解释了两者的区别及适用场景,前者适用于多分类单标签任务,后者则针对稀疏情况的多分类,y_true直接使用类别index。

在Keras中,官方内置了几种评价函数。

  • 对于二分类问题,评价指标可以用 binary_accuracy,就是最直观上讲的准确率。
  • 当面对多分类或者多标签的任务时,评价度量可能会用到这两个 categorical_accuracysparse_categorical_accuracy

binary_accuracy自然不必多讲,这篇文章讲一下categorical_accuracysparse_categorical_accuracy的区别:

Keras 官方API在这里, 里面没有讲到各个度量指标的具体含义,于是我们来看一下源码

def categorical_accuracy(y_true, y_pred):
    return K.cast(K.equal(K.argmax(y_true, axis=-1),
                          K.argmax(y_pred, axis=-1)),
                  K.floatx())

def sparse_categorical_accuracy(y_true, y_pred):
    return K.cast(K.equal(K.max(y_true, axis=-1),
                          K.cast(K.argmax(y_pred, axis=-1), K.floatx())),
                  K.floatx())

从代码可以看到:

categorical_accuracy:检查 y_ture 中最大值对应的index 与 y_pred 中最大值对应的index是否相等。

  • 注意,这里只比较一个值,即最大的那个值的index,这对于【多分类单标签】任务的是合适的,但并不适用于【多标签】任务。
  • 这里的 y_true 应为一个 one-hot 向量

sparse_categorical_accuracy检查 y_true 中的值(本身就是index) 与 y_pred 中最大值对应的index是否相等。

  • 针对稀疏情况的多分类,这里的 y_true 就是真实类的 index ,是个整数

举个栗子

如果有四个类,该样本属于第三类,那么在 categorical_accuracy 中 y_true =(0,0, 1, 0) , 而在 sparse_categorical_accuracy中 y_true = 2 (0-based计数)。但是,y_pred是一样的,均为softmax输出的vector,比如 y_pred = (0.02, 0.05, 0.83, 0.1),于是

y_true = (0, 0, 1, 0)
y_pred = (0.02, 0.05, 0.83, 0.1)
acc = categorical_accuracy(y_true, y_pred)
y_true = 2
y_pred = (0.02, 0.05, 0.83, 0.1)
acc = sparse_categorical_accuracy(y_true, y_pred)

Reference:
Keras - Difference between categorical_accuracy and sparse_categorical_accuracy

评论 11
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值