auc

    Area Under the Curve(AUC) Layer

    该层根据前向输出和标签计算AUC,在二分类(binary classification)估计中广泛使用。

    注:如果输入标注包含一种值,只有0或1两种情况,数据类型则强制转换成布尔值。相关定义可以在这里: 找到

    有两种可能的曲线:

    1. PR:准确率召回率曲线

    参数:

    • label (Tensor|LoDTensor) - 数据类型为int32,int64。二维整型变量,为训练数据的标签。

    • topk (int) - 取topk的输出值用于计算。

    • slide_steps (int) - 当计算batch auc时,不仅用当前步也用于先前步。slide_steps=1,表示用当前步;slide_steps = 3表示用当前步和前两步;slide_steps = 0,则用所有步。

    返回:元组,当前计算出的AUC。数据类型是tensor,支持float32和float64。 返回的元组为auc_out, batch_auc_out, [batch_stat_pos, batch_stat_neg, stat_pos, stat_neg]。 auc_out为准确率的结果。 batch_auc_out为batch准确率的结果。 batch_stat_pos为batch计算时label=1的统计值 batch_stat_neg为batch计算时label=0的统计值 stat_pos计算时label=1的统计值 stat_neg为计算时label=0的统计值