CommencerCommencer gratuitement

Recette simple

En utilisant le jeu de données attrition_num qui contient toutes les données numériques sur les employés ayant quitté l’entreprise, vous souhaitez construire un modèle capable de prédire si un employé est susceptible de rester, à partir de Attrition, une variable binaire codée comme un factor. Pour commencer, vous allez définir une recette simple qui ne fait rien d’autre que définir la formule du modèle et les données d’entraînement.

Les données attrition_num, la régression logistique lr_model, la fonction définie par l’utilisateur class-evaluate(), ainsi que les partitions train et test sont déjà chargées pour vous.

Cet exercice fait partie du cours

Feature engineering en R

Afficher le cours

Instructions

  • Créez une recette simple ne définissant que la formule du modèle.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Create a plain recipe defining only the model formula
lr_recipe_plain <- 
  ___(___ ~., ___ = ___)

lr_workflow_plain <- workflow() %>%
  add_model(lr_model) %>%
  add_recipe(lr_recipe_plain)
lr_fit_plain <- lr_workflow_plain %>%
  fit(train)
lr_aug_plain <-
  lr_fit_plain %>% augment(test)
lr_aug_plain %>% class_evaluate(truth = Attrition,
                 estimate = .pred_class,.pred_No)
Modifier et exécuter le code