Melatih dan menilai model
Sekarang setelah Anda menangani nilai hilang dan membuat variabel dummy, saatnya menilai kinerja model Anda!
Himpunan data attrition, beserta pemisahan test dan train, lr_recipe, serta logistic_model() yang telah Anda deklarasikan sudah dimuat untuk Anda.
Latihan ini adalah bagian dari kursus
Rekayasa Fitur di R
Petunjuk latihan
- Gabungkan model dan recipe dalam workflow.
- Latih workflow pada data train.
- Hasilkan data frame teraugmentasi untuk penilaian kinerja.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# 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))