Skapa en förväxlingsmatris
Som du såg i videon kan en förväxlingsmatris användas för att bedöma hur väl modellen presterar. Innan du skapar förväxlingsmatrisen behöver du dock klassificera de predikterade sannolikheterna som 1 eller 0 med hjälp av ett tröskelvärde.
Obs: 1 betyder Inaktiv och 0 betyder Aktiv.
prediction_test, som innehåller de predikterade sannolikheterna för personalomsättning för alla fall i test_set, finns tillgänglig i din arbetsmiljö.
Den här övningen är en del av kursen
HR-analys: Förutsäga personalomsättning i R
Övningsinstruktioner
- Omvandla de numeriska prediktorerna i
prediction_testtill en vektor med kategoriska prediktioner med ett tröskelvärde på 0,5. - Skapa förväxlingsmatrisen med hjälp av
prediction_categoriesoch de faktiska värdena i testuppsättningen (test_set$turnover).
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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