假设预测值的RDD为pred
实际真实值的RDD为realVal
我们计算AUC的公式为:
则AUC的计算可以按如下代码实现:
def aucCal(pred:RDD[(Long,Double)],realVal:RDD[(Long,Double)]):Double={
//join 操作
val pre=pred.zipWithIndex
val label=realVal.zipWithIndex
val combPair=pre.join(label)
//计算正样本的ran