开始使用免费开始使用

绘制 ROC 曲线

正如您在视频中所见,ROC 曲线可以高效地汇总分类器在所有可能阈值下的表现。这样就无需为许多不同阈值反复计算类别预测并逐一查看混淆矩阵,省去了大量繁琐工作。

我最喜欢用于计算 ROC 曲线的包是 caTools,其中包含一个名为 colAUC() 的函数。这个函数非常易用,还可以一次性为多个预测器计算 ROC 曲线。在本例中,您只需要为一个预测器计算 ROC 曲线,例如:

colAUC(predicted_probabilities, actual, plotROC = TRUE)

该函数会返回一个名为 AUC 的分数(稍后会详细介绍),并且设置 plotROC = TRUE 会绘制 ROC 曲线,方便可视化检查。

本练习是课程的一部分

在 R 中使用 caret 的机器学习

查看课程

练习说明

modeltesttrain(使用 sonar 数据)已从上一个练习加载到您的工作区。

  • 在测试集上预测概率(即 type = "response"),并将结果保存为 p
  • 使用测试集的预测概率绘制 ROC 曲线。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Predict on test: p


# Make ROC curve
编辑并运行代码