开始使用免费开始使用

创建混淆矩阵

正如视频中所示,混淆矩阵可以用来评估模型表现。在创建混淆矩阵之前,您需要先用一个阈值将预测概率分类为 10

注意:1 表示 Inactive,0 表示 Active。

prediction_test 已在您的工作区中,包含了 test_set 中所有样本的离职预测概率。

本练习是课程的一部分

HR Analytics:用 R 预测员工流失

查看课程

练习说明

  • 使用 0.5 作为阈值,将 prediction_test 中的数值型预测转换为分类型预测向量。
  • 使用 prediction_categories 与测试集中的真实值(test_set$turnover)创建混淆矩阵。

交互式实操练习

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

# Classify predictions using a cut-off of 0.5
prediction_categories <- ___(prediction_test > 0.5, 1, 0)

# Construct a confusion matrix
conf_matrix <- ___(prediction_categories, test_set$turnover)
conf_matrix
编辑并运行代码