创建混淆矩阵
正如视频中所示,混淆矩阵可以用来评估模型表现。在创建混淆矩阵之前,您需要先用一个阈值将预测概率分类为 1 或 0。
注意: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