开始使用免费开始使用

计算 ROC 曲线和 AUC

之前的练习已经表明,在类别不平衡的数据集上,仅用准确率衡量模型效果会产生很大误导。用图形展示模型性能,可以更直观地看到模型过于激进与过于保守之间的权衡。

在本练习中,您将创建 ROC 曲线并计算曲线下面积(AUC),以评估您先前构建的捐赠逻辑回归模型。

数据集 donors 已为您加载,其中包含预测概率列 donation_prob

本练习是课程的一部分

R 中的监督学习:分类

查看课程

练习说明

  • 加载 pROC 包。
  • 使用 roc(),并传入实际与预测捐赠的两列来创建 ROC 曲线。将结果保存为 ROC
  • 使用 plot() 绘制 ROC 对象。将曲线颜色指定为 col = "blue"
  • 使用 auc() 计算曲线下面积。

交互式实操练习

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

# Load the pROC package


# Create a ROC curve
ROC <- roc(___, ___)

# Plot the ROC curve
plot(___, col = ___)

# Calculate the area under the curve (AUC)
auc(___)
编辑并运行代码