开始使用免费开始使用

尝试另一个阈值

在前面的练习中,您使用了 0.50 作为阈值,将预测概率切分为类别预测(岩石 vs 地雷)。然而,这个分类阈值并不总是与具体建模问题的目标一致。

例如,假设您想识别那些您非常确定是地雷的目标。在这种情况下,您可能会使用 0.90 的概率阈值,以获得"更少的预测为地雷,但对每个预测更有把握"。

将概率切分成预测类别,然后计算混淆矩阵的代码模式,已在本章的练习 7 中展示。

本练习是课程的一部分

在 R 中使用 caret 的机器学习

查看课程

练习说明

  • 使用 ifelse() 创建字符向量 m_or_r:当 p 大于 0.9 时为正类 "M",否则为负类 "R"
  • m_or_r 转换为因子 p_class,其水平与 test[["Class"]] 的水平相同。
  • 使用 confusionMatrix() 构建混淆矩阵,传入 p_class 和数据集 test 中的 "Class" 列。

交互式实操练习

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

# If p exceeds threshold of 0.9, M else R: m_or_r


# Convert to factor: p_class


# Create confusion matrix
编辑并运行代码