Создание матрицы ошибок
Как вы видели в видео, матрица ошибок позволяет оценить качество работы модели. Однако прежде чем её построить, нужно классифицировать предсказанные вероятности как 1 или 0, используя пороговое значение.
Примечание: 1 означает «неактивен», 0 — «активен».
prediction_test, содержащий предсказанные вероятности увольнения для всех наблюдений из test_set, уже доступен в вашем рабочем пространстве.
Это упражнение является частью курса
HR-аналитика: прогнозирование текучести кадров на R
Инструкции к упражнению
- Преобразуйте числовые предсказания из
prediction_testв вектор категориальных предсказаний, используя пороговое значение 0,5. - Создайте матрицу ошибок на основе
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