Python tf.keras.metrics.Recall用法及代碼示例

計算關於標簽的預測的召回率。

繼承自：Metric、Layer、Module

用法

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

參數

thresholds (可選)[0, 1] 中的浮點值或浮點閾值的 python 列表/元組。將閾值與預測值進行比較以確定預測的真值(即，高於閾值的是true，低於閾值的是false)。為每個閾值生成一個度量值。如果未設置閾值和 top_k，則默認使用 thresholds=0.5 計算召回率。
top_k (可選)默認取消設置。一個 int 值，指定在計算召回時要考慮的 top-k 預測。
class_id (可選)我們想要二進製度量的整數類 ID。這必須在半開區間 [0, num_classes) 中，其中 num_classes 是預測的最後一個維度。
name (可選)指標實例的字符串名稱。
dtype (可選)度量結果的數據類型。

該指標創建兩個局部變量 true_positives 和 false_negatives ，用於計算召回率。該值最終返回為 recall ，這是一個冪等運算，隻需將 true_positives 除以 true_positives 和 false_negatives 的總和。

如果 sample_weight 是 None ，則權重默認為 1。使用 0 的 sample_weight 來屏蔽值。

如果設置了top_k，召回率將計算為批次條目標簽中的類在top-k預測中的平均頻率。

如果指定了class_id，我們通過僅考慮批次中class_id 位於標簽中的條目來計算召回率，並計算其中class_id 高於閾值和/或top-k 預測。

單機使用：

m = tf.keras.metrics.Recall()
m.update_state([0, 1, 1, 1], [1, 0, 1, 1])
m.result().numpy()
0.6666667

m.reset_state()
m.update_state([0, 1, 1, 1], [1, 0, 1, 1], sample_weight=[0, 0, 1, 0])
m.result().numpy()
1.0

compile() API 的用法：

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

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.metrics.Recall。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。