开始使用免费开始使用

计算 AUC

AUC 用于评估模型将观测从"成为目标的低概率"到"成为目标的高概率"正确排序的能力。在 Python 中,可以使用 roc_auc_score 函数计算模型的 AUC。它以目标的真实值和预测值作为参数。

您将先再次生成预测,然后计算其 roc_auc_score

本练习是课程的一部分

Python 预测分析入门

查看课程

练习说明

  • 上一章节中已为您创建并拟合了模型 logreg,DataFrame X 包含基表中的自变量列。请为基表中的样本生成预测。
  • 选择 predictions 的第 2 列,因为其中包含对目标的预测概率。
  • 目标的真实值已加载在 y 中。使用 roc_auc_score 函数计算模型的 AUC。

交互式实操练习

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

# Make predictions
predictions = logreg.____(____)
predictions_target = predictions[:,____]

# Calculate the AUC value
auc = ____(____, ____)
print(round(auc,2))
编辑并运行代码