Creează o matrice de confuzie
Așa cum ai văzut în video, o matrice de confuzie poate fi folosită pentru a determina cât de bine performează modelul tău. Înainte de a o crea, însă, trebuie să clasifici probabilitățile prezise ca 1 sau 0, folosind un prag de decizie.
Notă: 1 înseamnă Inactiv, iar 0 înseamnă Activ.
prediction_test, care conține probabilitățile prezise de fluctuație a angajaților pentru toate cazurile din test_set, este disponibil în spațiul tău de lucru.
Acest exercițiu face parte din cursul
HR Analytics: Predicția fluctuației angajaților în R
Instrucțiuni pentru exercițiu
- Transformă predicțiile numerice din
prediction_testîntr-un vector de predicții categoriale, folosind un prag de decizie de 0,5. - Creează matricea de confuzie folosind
prediction_categoriesși valorile reale din setul de testare (test_set$turnover).
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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