Box-Cox-transformation
Med datamängden attrition_num, som innehåller numeriska uppgifter 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 kvar. Målvariabeln är Attrition, en binär variabel kodad som en factor. För att få särdragen att bete sig ungefär normalfördelade skapar du ett recept som implementerar Box-Cox-transformationen.
Datamängden 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
Övningsinstruktioner
- Skapa ett recept som använder Box-Cox för att transformera alla numeriska särdrag, inklusive målvariabeln.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Create a recipe that uses Box-Cox to transform all numeric features
lr_recipe_BC <-
recipe(Attrition ~., data = train) %>%
___(___)
lr_workflow_BC <- workflow() %>%
add_model(lr_model) %>%
add_recipe(lr_recipe_BC)
lr_fit_BC <- lr_workflow_BC %>%
fit(train)
lr_aug_BC <-
lr_fit_BC %>% augment(test)
lr_aug_BC %>% class_evaluate(truth = Attrition,
estimate = .pred_class,.pred_No)