Kom igångKom igång gratis

Enkel recept

Med hjälp av datamängden attrition_num, som innehåller numeriska data om anställda som har lämnat företaget, vill du bygga en modell som kan förutsäga om en anställd sannolikt kommer att stanna. Du använder Attrition, en binär variabel kodad som en factor. Som första steg definierar du ett enkelt recept som inte gör annat än att ange modellformeln och träningsdatan.

Datan attrition_num, den logistiska regressionsmodellen lr_model, den användardefinierade funktionen class-evaluate() samt uppdelningarna train och test är redan inlästa.

Den här övningen är en del av kursen

Feature Engineering i R

Visa kurs

Övningsinstruktioner

  • Skapa ett enkelt recept som endast definierar modellformeln.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# 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)
Redigera och kör kod