Criar uma matriz de confusão
Como você viu no vídeo, uma matriz de confusão pode ser usada para avaliar o desempenho do seu modelo. Porém, antes de criar a matriz de confusão, você precisa classificar as probabilidades previstas como 1 ou 0, usando um ponto de corte.
Observação: 1 significa Inativo, enquanto 0 é Ativo.
prediction_test, que contém as probabilidades previstas de desligamento para todos os casos em test_set, está disponível no seu workspace.
Este exercicio faz parte do curso
Análise de RH: Prevendo Rotatividade de Funcionários em R
Instruções do exercicio
- Converta as previsões numéricas em
prediction_testem um vetor de previsões categóricas usando um ponto de corte de 0,5. - Crie a matriz de confusão usando
prediction_categoriese os valores reais no conjunto de teste (test_set$turnover).
exercicio interativo prático
Tente este exercicio completando este código de exemplo.
# 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