Začněte nyníZačněte zdarma

Transformace Box-Cox

Pracuješ s datasetem attrition_num, který obsahuje číselná data o zaměstnancích, kteří odešli z firmy. Tvým cílem je sestavit model predikující, zda zaměstnanec pravděpodobně zůstane, a to na základě proměnné Attrition – binární proměnné zakódované jako factor. Aby se příznaky chovaly přibližně normálně, vytvoříš recept implementující transformaci Box-Cox.

Data attrition_num, logistický regresní model lr_model, uživatelsky definovaná funkce class-evaluate() a rozdělení na train a test jsou již načteny.

Toto cvičení je součástí kurzu

Feature Engineering v R

Zobrazit kurz

Pokyny k cvičení

  • Vytvoř recept, který pomocí Box-Cox transformuje všechny numerické příznaky včetně cílové proměnné.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# 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)
Upravit a spustit kód