通往圆满的最后步骤

HR Analytics：用 R 预测员工流失

Anurag Gupta

People Analytics Practitioner

构建最终模型

# 最终模型，您将在下个练习中完成
final_log <- glm(...)
HR Analytics：用 R 预测员工流失

预测离职概率

# 为训练集生成预测
prediction_train <- predict(final_log, newdata = train_set_final, 
                            type = "response")
prediction_train[c(205, 645)]

       205        645 
0.06069079 0.99999898
HR Analytics：用 R 预测员工流失

绘制概率范围：训练集

# 查看预测概率范围
hist(prediction_train)

HR Analytics：用 R 预测员工流失

预测概率：测试集

# 为测试集生成预测
# test_set 是第 3 章练习 2 的测试数据集
prediction_test <- predict(final_log, newdata = test_set, 
                           type = "response")
HR Analytics：用 R 预测员工流失

绘制概率范围：测试集

# 查看预测概率范围
hist(prediction_test)

HR Analytics：用 R 预测员工流失

让我们练习！

HR Analytics：用 R 预测员工流失

Preparing Video For Download...