召回率(Recall)是衡量二元分类模型在正样本上的表现指标,其计算公式为:
其中,是真正例,
是假反例。
本题的一个小难点是对0的判断,不进行特判会导致除0错误。
标准代码如下
def recall(y_true, y_pred):
tp = np.sum((y_true == 1) & (y_pred == 1))
fn = np.sum((y_true == 1) & (y_pred == 0))
return round(tp / (tp + fn), 3) if tp + fn != 0 else 0.0