Keras 3 API 文档 / 指标 / 基于真/假正负样本的分类指标

基于真/假正负样本的分类指标

[来源]

AUC

keras.metrics.AUC(
    num_thresholds=200,
    curve="ROC",
    summation_method="interpolation",
    name=None,
    dtype=None,
    thresholds=None,
    multi_label=False,
    num_labels=None,
    label_weights=None,
    from_logits=False,
)

近似 ROC 或 PR 曲线的 AUC(曲线下面积)。

ROC(接收者操作特征,默认)或 PR(精确率召回率)曲线的 AUC(曲线下面积)是二元分类器的质量度量。与准确率不同,与交叉熵损失类似,ROC-AUC 和 PR-AUC 评估模型的所有操作点。

此类使用黎曼和近似 AUC。在指标累积阶段,预测根据其值累积在预定义的桶中。然后,通过插值每个桶的平均值来计算 AUC。这些桶定义了评估的操作点。

此指标创建四个局部变量:true_positivestrue_negativesfalse_positivesfalse_negatives,它们用于计算 AUC。为了离散化 AUC 曲线,使用线性间隔的阈值集来计算召回率和精确率值对。因此,ROC 曲线下的面积使用召回率值的高度乘以假阳性率来计算,而 PR 曲线下的面积使用精确率值的高度乘以召回率来计算。

此值最终作为 auc 返回,这是一个幂等操作,它使用上述变量计算的精确率与召回率值的离散化曲线的面积。num_thresholds 变量控制离散化的程度,阈值数量越大,越接近真实的 AUC。近似的质量可能因 num_thresholds 而异。thresholds 参数可用于手动指定阈值,这些阈值可以更均匀地分割预测。

为了最好地近似真实的 AUC,predictions 应该在 [0, 1] 范围内近似均匀分布(如果 from_logits=False)。如果不是这种情况,AUC 近似的质量可能会很差。将 summation_method 设置为 'minoring' 或 'majoring' 可以通过提供 AUC 的下限或上限估计来帮助量化近似误差。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

参数

  • num_thresholds: (可选)离散化 roc 曲线时使用的阈值数量。值必须大于 1。默认为 200
  • curve: (可选)指定要计算的曲线的名称,'ROC'(默认)或 'PR' 表示精确率召回率曲线。
  • summation_method: (可选)指定使用的 黎曼求和方法。'interpolation'(默认)对 ROC 应用中点求和方案。对于 PR-AUC,插值(真/假)阳性,但不是精度比率(有关详细信息,请参阅 Davis & Goadrich 2006);'minoring' 对递增的间隔应用左求和,对递减的间隔应用右求和;'majoring' 则相反。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。
  • thresholds: (可选)用于离散化曲线的阈值列表。如果设置,则忽略 num_thresholds 参数。值应该在 [0, 1] 中。端点阈值等于 {-epsilon, 1+epsilon}(对于一个小的正 epsilon 值)将自动包含在这些阈值中,以便正确处理等于 0 或 1 的预测。
  • multi_label: 布尔值,指示是否应该将多标签数据视为多标签数据,其中 AUC 是针对每个标签单独计算的,然后跨标签平均,或者(当 False 时)是否应该在 AUC 计算之前将数据展平为单个标签。在后一种情况下,当将多标签数据传递给 AUC 时,每个标签预测对都被视为一个单独的数据点。对于多类数据,应该设置为 False
  • num_labels: (可选)标签数量,在 multi_label 为 True 时使用。如果未指定 num_labels,则状态变量将在第一次调用 update_state 时创建。
  • label_weights: (可选)用于计算多标签数据的 AUC 的非负权重的列表、数组或张量。当 multi_label 为 True 时,权重应用于单个标签 AUC,并在它们平均以产生多标签 AUC 时应用。当它为 False 时,它们用于在计算扁平数据的混淆矩阵时对单个标签预测进行加权。请注意,这与 class_weights 不同,因为 class_weights 根据其标签的值对示例进行加权,而 label_weights 仅取决于该标签在扁平化之前的索引;因此,label_weights 不应用于多类数据。
  • from_logits: 布尔值,指示预测(update_state 中的 y_pred)是概率还是 sigmoid 对数。作为经验法则,当使用 keras 损失时,损失的 from_logits 构造函数参数应该与 AUC 的 from_logits 构造函数参数匹配。

示例

>>> m = keras.metrics.AUC(num_thresholds=3)
>>> m.update_state([0, 0, 1, 1], [0, 0.5, 0.3, 0.9])
>>> # threshold values are [0 - 1e-7, 0.5, 1 + 1e-7]
>>> # tp = [2, 1, 0], fp = [2, 0, 0], fn = [0, 1, 2], tn = [0, 2, 2]
>>> # tp_rate = recall = [1, 0.5, 0], fp_rate = [1, 0, 0]
>>> # auc = ((((1 + 0.5) / 2) * (1 - 0)) + (((0.5 + 0) / 2) * (0 - 0)))
>>> #     = 0.75
>>> m.result()
0.75
>>> m.reset_state()
>>> m.update_state([0, 0, 1, 1], [0, 0.5, 0.3, 0.9],
...                sample_weight=[1, 0, 0, 1])
>>> m.result()
1.0

compile() API 一起使用

# Reports the AUC of a model outputting a probability.
model.compile(optimizer='sgd',
              loss=keras.losses.BinaryCrossentropy(),
              metrics=[keras.metrics.AUC()])

# Reports the AUC of a model outputting a logit.
model.compile(optimizer='sgd',
              loss=keras.losses.BinaryCrossentropy(from_logits=True),
              metrics=[keras.metrics.AUC(from_logits=True)])

[来源]

Precision

keras.metrics.Precision(
    thresholds=None, top_k=None, class_id=None, name=None, dtype=None
)

计算预测相对于标签的精确率。

该指标创建两个局部变量:true_positivesfalse_positives,它们用于计算精确率。此值最终作为 precision 返回,这是一个幂等操作,它简单地将 true_positives 除以 true_positivesfalse_positives 的总和。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

如果设置了 top_k,我们将计算精确率,即批次条目中具有最高预测值的 top-k 类中,平均有多少类是正确的,并且可以在该条目的标签中找到。

如果指定了 class_id,我们将通过仅考虑批次中 class_id 超过阈值和/或位于 top-k 最高预测中的条目来计算精确率,并计算其中 class_id 确实是正确标签的条目的比例。

参数

  • thresholds: (可选)一个浮点值,或者一个 [0, 1] 中的浮点阈值列表/元组。将阈值与预测值进行比较以确定预测的真值(即,高于阈值是 True,低于阈值是 False)。如果与设置了 from_logits=True 的损失函数一起使用(即,没有将 sigmoid 应用于预测),则 thresholds 应设置为 0。为每个阈值生成一个指标值。如果既没有设置 thresholds 也没有设置 top_k,则默认情况下使用 thresholds=0.5 计算精确率。
  • top_k: (可选)默认情况下未设置。指定用于计算精确率时要考虑的 top-k 预测的整数。
  • class_id: (可选)我们要为其获取二元指标的整数类 ID。这必须在半开区间 [0, num_classes) 内,其中 num_classes 是预测的最后一个维度。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.Precision()
>>> m.update_state([0, 1, 1, 1], [1, 0, 1, 1])
>>> m.result()
0.6666667
>>> m.reset_state()
>>> m.update_state([0, 1, 1, 1], [1, 0, 1, 1], sample_weight=[0, 0, 1, 0])
>>> m.result()
1.0
>>> # With top_k=2, it will calculate precision over y_true[:2]
>>> # and y_pred[:2]
>>> m = keras.metrics.Precision(top_k=2)
>>> m.update_state([0, 0, 1, 1], [1, 1, 1, 1])
>>> m.result()
0.0
>>> # With top_k=4, it will calculate precision over y_true[:4]
>>> # and y_pred[:4]
>>> m = keras.metrics.Precision(top_k=4)
>>> m.update_state([0, 0, 1, 1], [1, 1, 1, 1])
>>> m.result()
0.5

compile() API 一起使用

model.compile(optimizer='sgd',
              loss='binary_crossentropy',
              metrics=[keras.metrics.Precision()])

与使用 from_logits=True 的损失一起使用

model.compile(optimizer='adam',
              loss=keras.losses.BinaryCrossentropy(from_logits=True),
              metrics=[keras.metrics.Precision(thresholds=0)])

[来源]

Recall

keras.metrics.Recall(
    thresholds=None, top_k=None, class_id=None, name=None, dtype=None
)

计算预测相对于标签的召回率。

此指标创建两个局部变量:true_positivesfalse_negatives,它们用于计算召回率。此值最终作为 recall 返回,这是一个幂等操作,它简单地将 true_positives 除以 true_positivesfalse_negatives 的总和。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

如果设置了 top_k,则召回率将计算为批次条目标签中平均有多少类位于 top-k 预测中。

如果指定了 class_id,我们将通过仅考虑批次中 class_id 位于标签中的条目来计算召回率,并计算其中 class_id 超过阈值和/或位于 top-k 预测中的条目的比例。

参数

  • thresholds: (可选)一个浮点值,或者一个 [0, 1] 中的浮点阈值列表/元组。将阈值与预测值进行比较以确定预测的真值(即,高于阈值是 True,低于阈值是 False)。如果与设置了 from_logits=True 的损失函数一起使用(即,没有将 sigmoid 应用于预测),则 thresholds 应设置为 0。为每个阈值生成一个指标值。如果既没有设置 thresholds 也没有设置 top_k,则默认情况下使用 thresholds=0.5 计算召回率。
  • top_k: (可选)默认情况下未设置。指定用于计算召回率时要考虑的 top-k 预测的整数。
  • class_id: (可选)我们要为其获取二元指标的整数类 ID。这必须在半开区间 [0, num_classes) 内,其中 num_classes 是预测的最后一个维度。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.Recall()
>>> m.update_state([0, 1, 1, 1], [1, 0, 1, 1])
>>> m.result()
0.6666667
>>> m.reset_state()
>>> m.update_state([0, 1, 1, 1], [1, 0, 1, 1], sample_weight=[0, 0, 1, 0])
>>> m.result()
1.0

compile() API 一起使用

model.compile(optimizer='sgd',
              loss='binary_crossentropy',
              metrics=[keras.metrics.Recall()])

与使用 from_logits=True 的损失一起使用

model.compile(optimizer='adam',
              loss=keras.losses.BinaryCrossentropy(from_logits=True),
              metrics=[keras.metrics.Recall(thresholds=0)])

[来源]

TruePositives

keras.metrics.TruePositives(thresholds=None, name=None, dtype=None)

计算真阳性数量。

如果给出 sample_weight,则计算真阳性的权重总和。此指标创建一个局部变量 true_positives,用于跟踪真阳性数量。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

参数

  • thresholds: (可选)默认为 0.5。一个浮点值,或者一个 [0, 1] 中的浮点阈值列表/元组。将阈值与预测值进行比较以确定预测的真值(即,高于阈值是 True,低于阈值是 False)。如果与设置了 from_logits=True 的损失函数一起使用(即,没有将 sigmoid 应用于预测),则 thresholds 应设置为 0。为每个阈值生成一个指标值。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.TruePositives()
>>> m.update_state([0, 1, 1, 1], [1, 0, 1, 1])
>>> m.result()
2.0
>>> m.reset_state()
>>> m.update_state([0, 1, 1, 1], [1, 0, 1, 1], sample_weight=[0, 0, 1, 0])
>>> m.result()
1.0

[来源]

TrueNegatives

keras.metrics.TrueNegatives(thresholds=None, name=None, dtype=None)

计算真阴性数量。

如果给出 sample_weight,则计算真阴性的权重总和。此指标创建一个局部变量 accumulator,用于跟踪真阴性数量。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

参数

  • thresholds: (可选)默认为 0.5。一个浮点值,或者一个 [0, 1] 中的浮点阈值列表/元组。将阈值与预测值进行比较以确定预测的真值(即,高于阈值是 True,低于阈值是 False)。如果与设置了 from_logits=True 的损失函数一起使用(即,没有将 sigmoid 应用于预测),则 thresholds 应设置为 0。为每个阈值生成一个指标值。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.TrueNegatives()
>>> m.update_state([0, 1, 0, 0], [1, 1, 0, 0])
>>> m.result()
2.0
>>> m.reset_state()
>>> m.update_state([0, 1, 0, 0], [1, 1, 0, 0], sample_weight=[0, 0, 1, 0])
>>> m.result()
1.0

[来源]

FalsePositives

keras.metrics.FalsePositives(thresholds=None, name=None, dtype=None)

计算假阳性数量。

如果给出 sample_weight,则计算假阳性的权重总和。此指标创建一个局部变量 accumulator,用于跟踪假阳性数量。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

参数

  • thresholds: (可选)默认为 0.5。一个浮点值,或者一个 [0, 1] 中的浮点阈值列表/元组。将阈值与预测值进行比较以确定预测的真值(即,高于阈值是 True,低于阈值是 False)。如果与设置了 from_logits=True 的损失函数一起使用(即,没有将 sigmoid 应用于预测),则 thresholds 应设置为 0。为每个阈值生成一个指标值。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.FalsePositives()
>>> m.update_state([0, 1, 0, 0], [0, 0, 1, 1])
>>> m.result()
2.0
>>> m.reset_state()
>>> m.update_state([0, 1, 0, 0], [0, 0, 1, 1], sample_weight=[0, 0, 1, 0])
>>> m.result()
1.0

[来源]

FalseNegatives

keras.metrics.FalseNegatives(thresholds=None, name=None, dtype=None)

计算假阴性数量。

如果给出 sample_weight,则计算假阴性的权重总和。此指标创建一个局部变量 accumulator,用于跟踪假阴性数量。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

参数

  • thresholds: (可选)默认为 0.5。一个浮点值,或者一个 [0, 1] 中的浮点阈值列表/元组。将阈值与预测值进行比较以确定预测的真值(即,高于阈值是 True,低于阈值是 False)。如果与设置了 from_logits=True 的损失函数一起使用(即,没有将 sigmoid 应用于预测),则 thresholds 应设置为 0。为每个阈值生成一个指标值。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.FalseNegatives()
>>> m.update_state([0, 1, 1, 1], [0, 1, 0, 0])
>>> m.result()
2.0
>>> m.reset_state()
>>> m.update_state([0, 1, 1, 1], [0, 1, 0, 0], sample_weight=[0, 0, 1, 0])
>>> m.result()
1.0

[来源]

PrecisionAtRecall

keras.metrics.PrecisionAtRecall(
    recall, num_thresholds=200, class_id=None, name=None, dtype=None
)

计算召回率大于等于指定值的最佳精确率。

此指标创建四个局部变量,true_positivestrue_negativesfalse_positivesfalse_negatives,它们用于计算给定召回率下的精确度。 给定召回值对应的阈值将被计算并用于评估相应的精确度。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

如果指定了 class_id,我们将通过仅考虑批次中 class_id 高于阈值预测的条目,并计算其中 class_id 确实是正确标签的条目的比例来计算精确度。

参数

  • 召回率:范围为 [0, 1] 的标量值。
  • num_thresholds:可选。默认值为 200。用于匹配给定召回率的阈值数量。
  • class_id: (可选)我们要为其获取二元指标的整数类 ID。这必须在半开区间 [0, num_classes) 内,其中 num_classes 是预测的最后一个维度。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.PrecisionAtRecall(0.5)
>>> m.update_state([0, 0, 0, 1, 1], [0, 0.3, 0.8, 0.3, 0.8])
>>> m.result()
0.5
>>> m.reset_state()
>>> m.update_state([0, 0, 0, 1, 1], [0, 0.3, 0.8, 0.3, 0.8],
...                sample_weight=[2, 2, 2, 1, 1])
>>> m.result()
0.33333333

compile() API 一起使用

model.compile(
    optimizer='sgd',
    loss='binary_crossentropy',
    metrics=[keras.metrics.PrecisionAtRecall(recall=0.8)])

[来源]

RecallAtPrecision

keras.metrics.RecallAtPrecision(
    precision, num_thresholds=200, class_id=None, name=None, dtype=None
)

计算精确度大于或等于指定值的最佳召回率。

对于给定的分数-标签-分布,可能无法实现所需的精确度,在这种情况下,将返回 0.0 作为召回率。

此指标创建四个局部变量,true_positivestrue_negativesfalse_positivesfalse_negatives,它们用于计算给定精确度下的召回率。 给定精确值对应的阈值将被计算并用于评估相应的召回率。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

如果指定了 class_id,我们将通过仅考虑批次中 class_id 高于阈值预测的条目,并计算其中 class_id 确实是正确标签的条目的比例来计算精确度。

参数

  • 精确度:范围为 [0, 1] 的标量值。
  • num_thresholds:可选。默认值为 200。用于匹配给定精确度的阈值数量。
  • class_id: (可选)我们要为其获取二元指标的整数类 ID。这必须在半开区间 [0, num_classes) 内,其中 num_classes 是预测的最后一个维度。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.RecallAtPrecision(0.8)
>>> m.update_state([0, 0, 1, 1], [0, 0.5, 0.3, 0.9])
>>> m.result()
0.5
>>> m.reset_state()
>>> m.update_state([0, 0, 1, 1], [0, 0.5, 0.3, 0.9],
...                sample_weight=[1, 0, 0, 1])
>>> m.result()
1.0

compile() API 一起使用

model.compile(
    optimizer='sgd',
    loss='binary_crossentropy',
    metrics=[keras.metrics.RecallAtPrecision(precision=0.8)])

[来源]

SensitivityAtSpecificity

keras.metrics.SensitivityAtSpecificity(
    specificity, num_thresholds=200, class_id=None, name=None, dtype=None
)

计算特异度大于或等于指定值的最佳敏感度。

敏感度 测量正确识别为阳性的实际阳性比例 (tp / (tp + fn))特异度 测量正确识别为阴性的实际阴性比例 (tn / (tn + fp))

此指标创建四个局部变量,true_positivestrue_negativesfalse_positivesfalse_negatives,它们用于计算给定特异度下的敏感度。 给定特异值对应的阈值将被计算并用于评估相应的敏感度。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

如果指定了 class_id,我们将通过仅考虑批次中 class_id 高于阈值预测的条目,并计算其中 class_id 确实是正确标签的条目的比例来计算精确度。

有关特异度和敏感度的更多信息,请参阅 以下内容

参数

  • 特异度:范围为 [0, 1] 的标量值。
  • num_thresholds:可选。默认值为 200。用于匹配给定特异度的阈值数量。
  • class_id: (可选)我们要为其获取二元指标的整数类 ID。这必须在半开区间 [0, num_classes) 内,其中 num_classes 是预测的最后一个维度。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.SensitivityAtSpecificity(0.5)
>>> m.update_state([0, 0, 0, 1, 1], [0, 0.3, 0.8, 0.3, 0.8])
>>> m.result()
0.5
>>> m.reset_state()
>>> m.update_state([0, 0, 0, 1, 1], [0, 0.3, 0.8, 0.3, 0.8],
...                sample_weight=[1, 1, 2, 2, 1])
>>> m.result()
0.333333

compile() API 一起使用

model.compile(
    optimizer='sgd',
    loss='binary_crossentropy',
    metrics=[keras.metrics.SensitivityAtSpecificity()])

[来源]

SpecificityAtSensitivity

keras.metrics.SpecificityAtSensitivity(
    sensitivity, num_thresholds=200, class_id=None, name=None, dtype=None
)

计算敏感度大于或等于指定值的最佳特异度。

敏感度 测量正确识别为阳性的实际阳性比例 (tp / (tp + fn))特异度 测量正确识别为阴性的实际阴性比例 (tn / (tn + fp))

此指标创建四个局部变量,true_positivestrue_negativesfalse_positivesfalse_negatives,它们用于计算给定敏感度下的特异度。 给定敏感值对应的阈值将被计算并用于评估相应的特异度。

如果 sample_weightNone,则权重默认为 1。使用 sample_weight 为 0 来屏蔽值。

如果指定了 class_id,我们将通过仅考虑批次中 class_id 高于阈值预测的条目,并计算其中 class_id 确实是正确标签的条目的比例来计算精确度。

有关特异度和敏感度的更多信息,请参阅 以下内容

参数

  • 敏感度:范围为 [0, 1] 的标量值。
  • num_thresholds:可选。默认值为 200。用于匹配给定敏感度的阈值数量。
  • class_id: (可选)我们要为其获取二元指标的整数类 ID。这必须在半开区间 [0, num_classes) 内,其中 num_classes 是预测的最后一个维度。
  • name: (可选)指标实例的字符串名称。
  • dtype: (可选)指标结果的数据类型。

示例

>>> m = keras.metrics.SpecificityAtSensitivity(0.5)
>>> m.update_state([0, 0, 0, 1, 1], [0, 0.3, 0.8, 0.3, 0.8])
>>> m.result()
0.66666667
>>> m.reset_state()
>>> m.update_state([0, 0, 0, 1, 1], [0, 0.3, 0.8, 0.3, 0.8],
...                sample_weight=[1, 1, 2, 2, 2])
>>> m.result()
0.5

compile() API 一起使用

model.compile(
    optimizer='sgd',
    loss='binary_crossentropy',
    metrics=[keras.metrics.SpecificityAtSensitivity()])

[来源]

F1Score

keras.metrics.F1Score(average=None, threshold=None, name="f1_score", dtype=None)

计算 F-1 分数。

公式

f1_score = 2 * (precision * recall) / (precision + recall)

这是精确度和召回率的调和平均值。 它的输出范围是 [0, 1]。 它适用于多类和多标签分类。

参数

  • average:对数据执行的平均类型。 可接受的值为 None"micro""macro""weighted"。 默认值为 None。 如果为 None,则不执行平均,result() 将返回每个类的分数。 如果为 "micro",则通过计算总的真正阳性、假阴性和假阳性来全局计算指标。 如果为 "macro",则计算每个标签的指标,并返回它们的非加权平均值。 这不考虑标签不平衡。 如果为 "weighted",则计算每个标签的指标,并返回通过支持(每个标签的真实实例数)加权的平均值。 这改变了 "macro" 以考虑标签不平衡。 它可能导致分数不在精确度和召回率之间。
  • threshold:大于 thresholdy_pred 元素将转换为 1,其余的转换为 0。 如果 thresholdNone,则 y_pred 的 argmax 将转换为 1,其余的转换为 0。
  • name:可选。指标实例的字符串名称。
  • dtype:可选。指标结果的数据类型。

返回

  • F-1 分数:浮点数。

示例

>>> metric = keras.metrics.F1Score(threshold=0.5)
>>> y_true = np.array([[1, 1, 1],
...                    [1, 0, 0],
...                    [1, 1, 0]], np.int32)
>>> y_pred = np.array([[0.2, 0.6, 0.7],
...                    [0.2, 0.6, 0.6],
...                    [0.6, 0.8, 0.0]], np.float32)
>>> metric.update_state(y_true, y_pred)
>>> result = metric.result()
array([0.5      , 0.8      , 0.6666667], dtype=float32)

[来源]

FBetaScore

keras.metrics.FBetaScore(
    average=None, beta=1.0, threshold=None, name="fbeta_score", dtype=None
)

计算 F-Beta 分数。

公式

b2 = beta ** 2
f_beta_score = (1 + b2) * (precision * recall) / (precision * b2 + recall)

这是精确度和召回率的加权调和平均值。 它的输出范围是 [0, 1]。 它适用于多类和多标签分类。

参数

  • average:在多类情况下跨每个类结果执行的平均类型。 可接受的值为 None"micro""macro""weighted"。 默认值为 None。 如果为 None,则不执行平均,result() 将返回每个类的分数。 如果为 "micro",则通过计算总的真正阳性、假阴性和假阳性来全局计算指标。 如果为 "macro",则计算每个标签的指标,并返回它们的非加权平均值。 这不考虑标签不平衡。 如果为 "weighted",则计算每个标签的指标,并返回通过支持(每个标签的真实实例数)加权的平均值。 这改变了 "macro" 以考虑标签不平衡。 它可能导致分数不在精确度和召回率之间。
  • beta:确定在精确度和召回率之间的调和平均值中赋予召回的权重(参见上面的伪代码方程)。 默认值为 1
  • threshold:大于 thresholdy_pred 元素将转换为 1,其余的转换为 0。 如果 thresholdNone,则 y_pred 的 argmax 将转换为 1,其余的转换为 0。
  • name:可选。指标实例的字符串名称。
  • dtype:可选。指标结果的数据类型。

返回

  • F-Beta 分数:浮点数。

示例

>>> metric = keras.metrics.FBetaScore(beta=2.0, threshold=0.5)
>>> y_true = np.array([[1, 1, 1],
...                    [1, 0, 0],
...                    [1, 1, 0]], np.int32)
>>> y_pred = np.array([[0.2, 0.6, 0.7],
...                    [0.2, 0.6, 0.6],
...                    [0.6, 0.8, 0.0]], np.float32)
>>> metric.update_state(y_true, y_pred)
>>> result = metric.result()
>>> result
[0.3846154 , 0.90909094, 0.8333334 ]