Dernières étapes vers le nirvana

Analytique RH : prédire le roulement du personnel avec R

Anurag Gupta

People Analytics Practitioner

Créez votre modèle final

# Modèle final : vous le compléterez dans le prochain exercice
final_log <- glm(...)

Prédire la probabilité d'attrition

# Prédire pour l'ensemble d'entraînement
prediction_train <- predict(final_log, newdata = train_set_final, 
                            type = "response")
prediction_train[c(205, 645)]

       205        645 
0.06069079 0.99999898

Tracer l'étendue des probabilités : ensemble d'entraînement

# Examiner l'étendue des prédictions
hist(prediction_train)

Prédire la probabilité : ensemble de test

# Prédire pour l'ensemble de test
# test_set est l'ensemble de test de l'exercice 2 du chapitre 3
prediction_test <- predict(final_log, newdata = test_set, 
                           type = "response")

Tracer l'étendue des probabilités : ensemble de test

# Examiner l'étendue des prédictions
hist(prediction_test)

Passons à la pratique !

Analytique RH : prédire le roulement du personnel avec R