Codificarea datelor categorice prin învățare supervizată

Ingineria caracteristicilor în R

Jorge Zazueta

Research Professor and Head of the Modeling Group at the School of Economics, UASLP

Introducere în codificarea supervizată

Codificarea supervizată utilizează valorile rezultatului pentru a extrage caracteristici numerice din predictori nominali.

Introducere în codificarea supervizată

Codificarea supervizată utilizează valorile rezultatului pentru a extrage caracteristici numerice din predictori nominali.

Câteva funcții de codificare supervizată disponibile în pachetul embed

Funcție	Definiție
step_lencode_glm()	Utilizează codificări de verosimilitate pentru a converti un predictor nominal într-un set de scoruri derivate dintr-un model liniar generalizat.
step_lencode_bayes()	Aplică codificări bayesiene de verosimilitate pentru a converti un predictor nominal într-un set de scoruri derivate dintr-un model liniar generalizat estimat prin analiză bayesiană.
step_lencode_mixed()	Convertește predictorii nominali într-un set de scoruri derivate dintr-un model liniar generalizat mixt.

Prezicerea succesului cererilor de grant

Dorim să prezicem succesul cererilor de grant exclusiv pe baza codului sponsorului.

lr_model <- logistic_reg() # declare model

lr_recipe_glm <- # Set recipe glm
  recipe(class ~ sponsor_code, 
         data = grants_train) %>%
  step_lencode_glm(sponsor_code, 
                   # Declare outcome variable
                   outcome = vars(class))

lr_workflow_glm <- # Create Workflow
  workflow() %>%
  add_model(lr_model) %>%
  add_recipe(lr_recipe_glm)

Rezumatul fluxului de lucru

lr_workflow_glm

-- Workflow ------------------------------------
Preprocessor: Recipe
Model: logistic_reg()

-- Preprocessor --------------------------------
1 Recipe Step

- step_lencode_glm()

-- Model --------------------------------------
Logistic Regression Model Specification (classification)

Computational engine: glm

Antrenare, augmentare și evaluare

Antrenăm și evaluăm modelul

lr_fit_glm <- # Fit
  lr_workflow_glm %>%
  fit(grants_train)

lr_aug_glm <- # Augment
  lr_fit_glm %>%
  augment(grants_test)

glm_model <- lr_aug_glm %>% # Assess
  class_evaluate(truth = class,
                 estimate = .pred_class,
                 .pred_successful)

Rezultatele de performanță sunt stocate în glm_model

glm_model

# A tibble: 2 × 3
  .metric  .estimator .estimate
  <chr>    <chr>          <dbl>
1 accuracy binary         0.728
2 roc_auc  binary         0.684

Combinarea modelelor

Construim bayes_model și mixed_model pentru a compara performanța pașilor corespunzători.

# Define model names
model <- c("glm", "glm",
           "bayes","bayes",
           "mixed", "mixed")
# Bind models in a tibble
models <- 
bind_rows(glm_model,
          bayes_model,
          mixed_model)%>%
  add_column(model = model)%>%
  select(-.estimator) %>%
  spread(model,.estimate)

Un tabel de performanță convenabil

models

# A tibble: 2 × 4
  .metric  bayes   glm mixed
  <chr>    <dbl> <dbl> <dbl>
1 accuracy 0.718 0.728 0.720
2 roc_auc  0.686 0.684 0.682

Vizualizarea rezultatelor

Vizualizați rezultatele într-un grafic cu coordonate paralele din pachetul Gally.

# Libraries
library(GGally)

# Parallel coordinates chart
ggparcoord(models,
           columns = 2:4, 
           groupColumn = 1,
           scale="globalminmax",
           showPoints = TRUE)

Grafic cu coordonate paralele pentru accuracy și roc_auc, comparând toate modelele.

Grafic cu coordonate paralele pentru accuracy și roc_auc, comparând toate modelele.

Să exersăm!

Ingineria caracteristicilor în R