1. Learn
  2. /
  3. 课程
  4. /
  5. 使用 scikit-learn 的监督学习

Connected

道练习

ROC 曲线

现在,您已经构建了用于预测糖尿病状态的逻辑回归模型,可以绘制 ROC 曲线,直观展示当决策阈值改变时,真正率和假正率如何变化。

测试标签 y_test,以及测试特征属于正类的预测概率 y_pred_probs 已为您预加载,同时 matplotlib.pyplot 已以 plt 导入。

您将创建一条 ROC 曲线,并对结果进行解读。

说明 1 / 共 2 个

undefined XP
    1
    2
  • 导入 roc_curve。
  • 使用 y_test 和 y_pred_probs 计算 ROC 曲线的取值,并将结果解包到 fpr、tpr 和 thresholds。
  • 绘制真正率与假正率的关系图。