Modelagem com tidymodels em R
David Svancer
Data Scientist

Definir papéis das colunas
Definir tipos de dados das variáveis
Feito com a função recipe()

Adicionar etapas de pré-processamento
Cada etapa é adicionada com uma função step_*() específica

Objetos recipe são treinados em uma fonte de dados, tipicamente o conjunto de treino
Recipes são treinadas com a função prep()

Aplicar todas as transformações treinadas
Recipes são aplicadas com a função bake()

Aplicar log em total_time nos dados de lead scoring
leads_training
# A tibble: 996 x 7
purchased total_visits total_time pages_per_visit total_clicks lead_source us_location
<fct> <dbl> <dbl> <dbl> <dbl> <fct> <fct>
1 yes 7 1148 7 59 direct_traffic west
2 no 5 228 2.5 25 email southeast
3 no 7 481 2.33 21 organic_search west
4 no 4 177 4 37 direct_traffic west
5 no 2 1273 2 26 email midwest
# ... with 991 more rows
A função recipe()
data
Passe o objeto recipe para step_log() para adicionar a transformação logarítmica
total_time, e especifique a base do logleads_log_rec <- recipe(purchased ~ ., data = leads_training) %>%step_log(total_time, base = 10)
leads_log_rec
Data Recipe
Inputs:
role #variables
outcome 1
predictor 6
Operations:
Log transformation on total_time
Passar um objeto recipe para a função summary()
typeroleleads_log_rec %>%
summary()
# A tibble: 7 x 4
variable type role source
<chr> <chr> <chr> <chr>
1 total_visits numeric predictor original
2 total_time numeric predictor original
3 pages_per_visit numeric predictor original
4 total_clicks numeric predictor original
5 lead_source nominal predictor original
6 us_location nominal predictor original
7 purchased nominal outcome original
A função prep()
recipe como primeiro argumentotraining
Imprimindo um objeto recipe treinado
[trained]leads_log_rec_prep <- leads_log_rec %>%
prep(training = leads_training)
leads_log_rec_prep
Data Recipe
Inputs:
role #variables
outcome 1
predictor 6
Training data contained 996 data points and
no missing data.
Operations:
Log transformation on total_time [trained]
A função bake()
recipe treinadonew_datarecipe treinadaleads_training foi usado para treinar a recipeprep()NULL em new_data para extrairleads_log_rec_prep %>%
bake(new_data = NULL)
# A tibble: 996 x 7
total_visits total_time ... us_location purchased
<dbl> <dbl> ... <fct> <fct>
1 7 3.06 ... west yes
2 5 2.36 ... southeast no
3 7 2.68 ... west no
4 4 2.25 ... west no
5 2 3.10 ... midwest no
# ... with 991 more rows
Transformando conjuntos não usados no treino da recipe
new_datarecipe treinada aplicará todas as etapas aos novos dadosleads_log_rec_prep %>%
bake(new_data = leads_test)
# A tibble: 332 x 7
total_visits total_time ... us_location purchased
<dbl> <dbl> ... <fct> <fct>
1 8 2 ... west no
2 4 3.13 ... northeast yes
3 3 2.25 ... west no
4 2 1.20 ... midwest no
5 9 3.01 ... west yes
# ... with 327 more rows
Modelagem com tidymodels em R