使用包ROCR的精确回收曲线的AUC

时间:2016-09-01 09:47:03

标签: r auc precision-recall

如何使用ROCR包获得精密回收曲线的AUC(曲线下面积)?

library(ROCR)
data(ROCR.simple)
pred <- prediction( ROCR.simple$predictions, ROCR.simple$labels)
perf <- performance(pred,"tpr","fpr")
plot(perf)
## precision/recall curve (x-axis: recall, y-axis: precision)
perf1 <- performance(pred, "prec", "rec")
plot(perf1)

2 个答案:

答案 0 :(得分:1)

您可以先获得精确值和召回值

x <- perf1@x.values[[1]] # Recall values
y <- perf1@y.values[[1]] # Precision values

然后使用任何方法计算曲线下面积 calculate area under the curve

答案 1 :(得分:0)

ROCR可以直接计算AUC:

perf <- performance(pred, "auc")

获取AUC

perf@y.values[[1]]