Xác thực kết quả hồi quy logistic

Phân tích nhân sự: Dự đoán nghỉ việc bằng R

Anurag Gupta

People Analytics Practitioner

Phân phối xác suất nghỉ việc của tập kiểm tra

Chuyển xác suất thành nhóm bằng ngưỡng cắt

# Phân loại dự đoán với ngưỡng 0.5
pred_cutoff_50_test <- ifelse(predictions_test > 0.5, 1, 0)

Confusion matrix là gì?

Confusion matrix đo lường hiệu suất của mô hình phân loại.

Tạo confusion matrix

## Tạo confusion matrix
table(pred_cutoff_50_test, test_set$turnover)

prediction_categories   0   1
                    0 450  22
                    1  20  94

Hiểu về confusion matrix

True negatives (TN): Mô hình xác định đúng nhân viên đang hoạt động
True positives (TP): Mô hình xác định đúng nhân viên đã nghỉ
False positives (FP): Mô hình dự đoán nhân viên đã nghỉ, nhưng thực tế đang hoạt động
False negatives (FN): Mô hình dự đoán nhân viên đang hoạt động, nhưng thực tế đã nghỉ

Confusion matrix: accuracy

$$ \text{Accuracy} = \frac{\text{ TP + TN }}{\text{ TP + TN + FP + FN }} $$

$$ \text{Accuracy} = \frac{\text{ 450 + 94 }}{\text{ 450 + 94 + 22 + 20}} $$

$$ = $$

$$ \text{0.9283} $$

Tạo confusion matrix

# Nạp thư viện
library(caret)

# Tạo confusion matrix
conf_matrix_50 <- confusionMatrix(table(test_set$turnover, 
                                        pred_cutoff_50_test))

conf_matrix_50

Confusion Matrix and Statistics

prediction_categories   0   1
                    0 450  22
                    1  20  94

               Accuracy : 0.9283          
                 95% CI : (0.9044, 0.9479)
    No Information Rate : 0.802           
    P-Value [Acc > NIR] : <2e-16                                        
                  Kappa : 0.7728          
 Mcnemar's Test P-Value : 0.8774                                       
            Sensitivity : 0.9574          
            Specificity : 0.8103          
         Pos Pred Value : 0.9534          
         Neg Pred Value : 0.8246          
             ...

Tài nguyên cho phương pháp nâng cao

Cùng luyện tập nào!

Phân tích nhân sự: Dự đoán nghỉ việc bằng R