Crea una matriz de confusión
Como viste en el vídeo, una matriz de confusión sirve para evaluar qué tal está rindiendo tu modelo. Pero antes de crearla, necesitas clasificar las probabilidades previstas como 1 o 0 usando un umbral.
Nota: 1 significa Inactive y 0 es Active.
prediction_test, que contiene las probabilidades previstas de abandono para todos los casos en test_set, está disponible en tu espacio de trabajo.
Este ejercicio forma parte del curso
Analítica de RR. HH.: Predicción de la rotación de empleados en R
Instrucciones del ejercicio
- Convierte las predicciones numéricas de
prediction_testen un vector de predicciones categóricas usando un umbral de 0.5. - Crea la matriz de confusión usando
prediction_categoriesy los valores reales del conjunto de prueba (test_set$turnover).
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# 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