Tạo ma trận nhầm lẫn
Như bạn đã thấy trong video, ma trận nhầm lẫn dùng để đánh giá mức độ hoạt động của mô hình. Tuy nhiên, trước khi tạo ma trận nhầm lẫn, bạn cần phân loại các xác suất dự đoán thành 1 hoặc 0 bằng một ngưỡng (cut-off).
Lưu ý: 1 là Inactive, 0 là Active.
prediction_test, chứa các xác suất dự đoán nghỉ việc cho tất cả các trường hợp trong test_set, đã có sẵn trong không gian làm việc của bạn.
Bài tập này là một phần của khóa học
Phân tích nhân sự: Dự đoán nghỉ việc bằng R
Hướng dẫn bài tập
- Chuyển các dự đoán dạng số trong
prediction_testthành một vector dự đoán dạng phân loại với ngưỡng 0.5. - Tạo ma trận nhầm lẫn bằng
prediction_categoriesvà giá trị thực tế trong tập kiểm tra (test_set$turnover).
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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