ÎncepețiÎncepe gratuit

Antrenarea și evaluarea modelului

Acum că ai rezolvat valorile lipsă și ai creat variabile dummy, este momentul să evaluezi performanța modelului!

Setul de date attrition, împreună cu împărțirile test și train, rețeta lr_recipe și modelul logistic_model() declarat sunt deja încărcate pentru tine.

Acest exercițiu face parte din cursul

Ingineria caracteristicilor în R

Vezi cursul

Instrucțiuni pentru exercițiu

  • Grupează modelul și rețeta într-un workflow.
  • Antrenează workflow-ul pe datele de antrenament.
  • Generează un data frame augmentat pentru evaluarea performanței.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Bundle model and recipe in workflow
lr_workflow <- ___() %>%
  add_model(lr_model) %>%
  add_recipe(lr_recipe)

# Fit workflow to the train data
lr_fit <- ___(lr_workflow, data = train)

# Generate an augmented data frame for performance assessment
lr_aug <- lr_fit %>% ___(test)

lr_aug %>% roc_curve(truth = Attrition, .pred_No) %>% autoplot()
bind_rows(lr_aug %>% roc_auc(truth = Attrition, .pred_No),          
          lr_aug %>% accuracy(truth = Attrition, .pred_class))
Editează și rulează codul