Introduction to machine learning

Lecture 16

Dr. Benjamin Soltoff

Cornell University
INFO 5001 - Fall 2026

October 22, 2026

Announcements

Announcements

• Sign up for a quiz 02 interview time slot
• Project proposal feedback

Learning objectives

• Review the major goals of machine learning
• Introduce {tidymodels} and the {parsnip} packages for estimating regression models
• Define resampling methods for evaluating model performance
• Implement cross-validation using {rsample}

Application exercise

ae-14

Instructions

• Go to the course GitHub org and find your ae-14 (repo name will be suffixed with your GitHub name).
• Clone the repo in Positron, run renv::restore() to install the required packages, open the Quarto document in the repo, and follow along and complete the exercises.
• Render, commit, and push your edits by the AE deadline – end of the day

What is machine learning?

Examples of supervised learning

• Will a user click on this ad?
• Will this property flood in the next year?
• Will a police officer engage in misconduct in the next six months?
• How many individuals will become infected with COVID-19 in the next week?
• How long until Denmark’s parliamentary coalition collapses?

Two modes

Classification

Will this home sell in the next 30 days?

Regression

What will the sale price be for this home?

Two cultures

Statistics

• model first
• inference emphasis

Machine Learning

• data first
• prediction emphasis

Statistics

“Statisticians, like artists, have the bad habit of falling in love with their models.”

— George Box

{tidymodels}

{tidymodels}

Predictive modeling

library(tidymodels)
## ── Attaching packages ────────────────────────────────────────────────────────── tidymodels 1.5.0 ──
## ✔ broom        1.0.13     ✔ rsample      1.3.2 
## ✔ dials        1.4.4      ✔ tailor       0.1.0 
## ✔ infer        1.1.0      ✔ tune         2.1.0 
## ✔ modeldata    1.5.1      ✔ workflows    1.3.0 
## ✔ parsnip      1.6.0      ✔ workflowsets 1.1.1 
## ✔ recipes      1.3.3      ✔ yardstick    1.4.0
## ── Conflicts ───────────────────────────────────────────────────────────── tidymodels_conflicts() ──
## ✖ scales::discard() masks purrr::discard()
## ✖ dplyr::filter()   masks stats::filter()
## ✖ recipes::fixed()  masks stringr::fixed()
## ✖ dplyr::lag()      masks stats::lag()
## ✖ yardstick::spec() masks readr::spec()
## ✖ recipes::step()   masks stats::step()

Children at hotels

Hotel booking demand

library(tidymodels)

hotels <- read_csv("data/hotels-lite.csv") |>
  mutate(across(where(is.character), as.factor))

dim(hotels)
## [1] 4000   22

Hotel booking demand

• N = 4000
• 1 categorical outcome: children
• 21 predictors

What is the goal of machine learning?

Build models that

generate accurate predictions

for future, yet-to-be-seen data.

Machine learning

We’ll use this goal to drive learning of 3 core {tidymodels} packages:

• {parsnip}
• {rsample}
• {yardstick}

🔨 Build models with {parsnip}

{parsnip}

To specify a model with {parsnip}

1. Pick a model
2. Set the engine
3. Set the mode (if needed)

To specify a model with {parsnip}

logistic_reg() |>
  set_engine("glm") |>
  set_mode("classification")
## Logistic Regression Model Specification (classification)
## 
## Computational engine: glm

To specify a model with {parsnip}

decision_tree() |>
  set_engine("C5.0") |>
  set_mode("classification")
## Decision Tree Model Specification (classification)
## 
## Computational engine: C5.0

To specify a model with {parsnip}

nearest_neighbor() |>
  set_engine("kknn") |>
  set_mode("classification")
## K-Nearest Neighbor Model Specification (classification)
## 
## Computational engine: kknn

1. Pick a model

logistic_reg()

Specifies a model that uses logistic regression

logistic_reg(penalty = NULL, mixture = NULL)

logistic_reg()

Specifies a model that uses logistic regression

logistic_reg(
  mode = "classification", # "default" mode, if exists
  penalty = NULL, # model hyper-parameter
  mixture = NULL # model hyper-parameter
)

set_engine()

Adds an engine to power or implement the model.

logistic_reg() |> set_engine(engine = "glm")

Set the engine when you define the model type.

logistic_reg(engine = "glm")

set_mode()

Sets the class of problem the model will solve, which influences which output is collected. Not necessary if mode is set in Step 1.

logistic_reg() |> set_mode(mode = "classification")

⌨️ Specify a model

Instructions

Run the chunk in your .qmd and look at the output. Then, copy/paste the code and edit to create:

• a decision tree model for classification

• that uses the {C5.0} engine.

Save it as tree_mod and look at the object. What is different about the output?

Hint: you’ll need https://www.tidymodels.org/find/parsnip/

02:00

lr_mod
## Logistic Regression Model Specification (classification)
## 
## Computational engine: glm

tree_mod <- decision_tree() |>
  set_engine(engine = "C5.0") |>
  set_mode("classification")
tree_mod
## Decision Tree Model Specification (classification)
## 
## Computational engine: C5.0

Use a model

Now we’ve built a model.

But, how do we use a model?

First - what does it mean to use a model?

Learn from data

Statistical models learn from the data.

Many learn model parameters, which can be useful as values for inference and interpretation.

fit()

Train a model by fitting a model. Returns a parsnip model fit.

fit(
  tree_mod,
  children ~ average_daily_rate + stays_in_weekend_nights,
  data = hotels
)

fit()

Train a model by fitting a model. Returns a parsnip model fit.

tree_mod |>
  fit(
    children ~ average_daily_rate + stays_in_weekend_nights,
    data = hotels
  )

A fitted model

lr_mod |>
  fit(children ~ average_daily_rate + stays_in_weekend_nights, data = hotels) |>
  broom::tidy()
## # A tibble: 3 × 5
##   term                estimate std.error statistic   p.value
##   <chr>                  <dbl>     <dbl>     <dbl>     <dbl>
## 1 (Intercept)           2.06    0.0956       21.5  5.31e-103
## 2 average_daily_rate   -0.0168  0.000710    -23.7  9.78e-124
## 3 stays_in_weekend_n…  -0.0671  0.0352       -1.91 5.63e-  2

“All models are wrong, but some are useful”

##           Truth
## Prediction children none
##   children     1277  483
##   none          723 1517

“All models are wrong, but some are useful”

##           Truth
## Prediction children none
##   children     1446  646
##   none          554 1354

Axiom

The best way to measure a model’s performance at predicting new data is to predict new data.

♻️ Resample models with {rsample}

{rsample}

The holdout method

initial_split()*

“Splits” data randomly into a single testing and a single training set.

initial_split(data, prop = 3 / 4)

* from {rsample}

initial_split()

hotels_split <- initial_split(data = hotels, prop = 3 / 4)
hotels_split
## <Training/Testing/Total>
## <3000/1000/4000>

training() and testing()*

Extract training and testing sets from an rsplit

training(hotels_split)
testing(hotels_split)

* from {rsample}

training()

hotels_train <- training(hotels_split)
hotels_train
## # A tibble: 3,000 × 22
##    hotel        lead_time stays_in_weekend_nights
##    <fct>            <dbl>                   <dbl>
##  1 City_Hotel          65                       1
##  2 City_Hotel           0                       2
##  3 Resort_Hotel       119                       2
##  4 City_Hotel         302                       2
##  5 City_Hotel         157                       1
##  6 Resort_Hotel        42                       1
##  7 City_Hotel          36                       0
##  8 City_Hotel          87                       1
##  9 City_Hotel           6                       2
## 10 City_Hotel         108                       0
## # ℹ 2,990 more rows
## # ℹ 19 more variables: stays_in_week_nights <dbl>,
## #   adults <dbl>, children <fct>, meal <fct>,
## #   market_segment <fct>, …

⌨️ Split the data

Instructions

Fill in the blanks.

Use initial_split(), training(), and testing() to:

1. Split hotels into training and test sets. Save the rsplit!

2. Extract the training data and testing data.

3. Check the proportions of the children variable in each set.

Keep set.seed(100) at the start of your code.

04:00

set.seed(100) # Important!

hotels_split <- initial_split(hotels, prop = 3 / 4)
hotels_train <- training(hotels_split)
hotels_test <- testing(hotels_split)

count(x = hotels_train, children) |>
  mutate(prop = n / sum(n))
## # A tibble: 2 × 3
##   children     n  prop
##   <fct>    <int> <dbl>
## 1 children  1503 0.501
## 2 none      1497 0.499
count(x = hotels_test, children) |>
  mutate(prop = n / sum(n))
## # A tibble: 2 × 3
##   children     n  prop
##   <fct>    <int> <dbl>
## 1 children   497 0.497
## 2 none       503 0.503

Note: Stratified sampling

Use stratified random sampling to ensure identical proportions in training/test sets

set.seed(100) # Important!

hotels_split <- initial_split(hotels, prop = 3 / 4, strata = children)
hotels_train <- training(hotels_split)
hotels_test <- testing(hotels_split)

count(x = hotels_train, children) |>
  mutate(prop = n / sum(n))
## # A tibble: 2 × 3
##   children     n  prop
##   <fct>    <int> <dbl>
## 1 children  1500   0.5
## 2 none      1500   0.5
count(x = hotels_test, children) |>
  mutate(prop = n / sum(n))
## # A tibble: 2 × 3
##   children     n  prop
##   <fct>    <int> <dbl>
## 1 children   500   0.5
## 2 none       500   0.5

Data Splitting

Take the average!

The testing set is precious…

we can only use it once!

How can we use the training set to compare, evaluate, and tune models?

Cross-validation

V-fold cross-validation

vfold_cv(data, v = 10, ...)

Guess

How many times does an observation/row appear in the assessment set?

Quiz

If we use 10 folds, which percent of our data will end up in the analysis set and which percent in the assessment set for each fold?

90% - analysis

10% - assessment

⌨️ Generate cross-validation folds

Instructions

Run the code below. What does it return?

set.seed(100)
hotels_folds <- vfold_cv(data = hotels_train, v = 10)
hotels_folds

01:00

set.seed(100)
hotels_folds <- vfold_cv(data = hotels_train, v = 10)
hotels_folds
## #  10-fold cross-validation 
## # A tibble: 10 × 2
##    splits             id    
##    <list>             <chr> 
##  1 <split [2700/300]> Fold01
##  2 <split [2700/300]> Fold02
##  3 <split [2700/300]> Fold03
##  4 <split [2700/300]> Fold04
##  5 <split [2700/300]> Fold05
##  6 <split [2700/300]> Fold06
##  7 <split [2700/300]> Fold07
##  8 <split [2700/300]> Fold08
##  9 <split [2700/300]> Fold09
## 10 <split [2700/300]> Fold10

fit_resamples()

Trains and tests a resampled model.

tree_mod |>
  fit_resamples(
    children ~ average_daily_rate + stays_in_weekend_nights,
    resamples = hotels_folds
  )

tree_mod |>
  fit_resamples(
    children ~ average_daily_rate + stays_in_weekend_nights,
    resamples = hotels_folds
  )
## # Resampling results
## # 10-fold cross-validation 
## # A tibble: 10 × 4
##    splits             id     .metrics         .notes  
##    <list>             <chr>  <list>           <list>  
##  1 <split [2700/300]> Fold01 <tibble [3 × 4]> <tibble>
##  2 <split [2700/300]> Fold02 <tibble [3 × 4]> <tibble>
##  3 <split [2700/300]> Fold03 <tibble [3 × 4]> <tibble>
##  4 <split [2700/300]> Fold04 <tibble [3 × 4]> <tibble>
##  5 <split [2700/300]> Fold05 <tibble [3 × 4]> <tibble>
##  6 <split [2700/300]> Fold06 <tibble [3 × 4]> <tibble>
##  7 <split [2700/300]> Fold07 <tibble [3 × 4]> <tibble>
##  8 <split [2700/300]> Fold08 <tibble [3 × 4]> <tibble>
##  9 <split [2700/300]> Fold09 <tibble [3 × 4]> <tibble>
## 10 <split [2700/300]> Fold10 <tibble [3 × 4]> <tibble>

collect_metrics()

Unnest the metrics column from a tidymodels fit_resamples()

_results |> collect_metrics(summarize = TRUE)

TRUE is actually the default; averages across folds

tree_fit <- tree_mod |>
  fit_resamples(
    children ~ average_daily_rate + stays_in_weekend_nights,
    resamples = hotels_folds
  )
collect_metrics(tree_fit)
## # A tibble: 3 × 6
##   .metric     .estimator  mean     n std_err .config        
##   <chr>       <chr>      <dbl> <int>   <dbl> <chr>          
## 1 accuracy    binary     0.696    10 0.0105  pre0_mod0_post0
## 2 brier_class binary     0.209    10 0.00424 pre0_mod0_post0
## 3 roc_auc     binary     0.697    10 0.0104  pre0_mod0_post0

collect_metrics(tree_fit, summarize = FALSE)
## # A tibble: 30 × 5
##    id     .metric     .estimator .estimate .config        
##    <chr>  <chr>       <chr>          <dbl> <chr>          
##  1 Fold01 accuracy    binary         0.677 pre0_mod0_post0
##  2 Fold01 roc_auc     binary         0.670 pre0_mod0_post0
##  3 Fold01 brier_class binary         0.221 pre0_mod0_post0
##  4 Fold02 accuracy    binary         0.73  pre0_mod0_post0
##  5 Fold02 roc_auc     binary         0.731 pre0_mod0_post0
##  6 Fold02 brier_class binary         0.194 pre0_mod0_post0
##  7 Fold03 accuracy    binary         0.683 pre0_mod0_post0
##  8 Fold03 roc_auc     binary         0.698 pre0_mod0_post0
##  9 Fold03 brier_class binary         0.209 pre0_mod0_post0
## 10 Fold04 accuracy    binary         0.71  pre0_mod0_post0
## # ℹ 20 more rows

10-fold CV

• 10 different analysis/assessment sets

• 10 different models (trained on analysis sets)

• 10 different sets of performance statistics (on assessment sets)

📏 Evaluate models with {yardstick}

{yardstick}

roc_curve()

Takes predictions from a special kind of fit_resamples().

Returns a tibble with probabilities.

roc_curve(data, truth, ...)

truth = actual outcome

... = predicted probability of outcome

tree_preds <- tree_mod |>
  fit_resamples(
    children ~ average_daily_rate + stays_in_weekend_nights,
    resamples = hotels_folds,
    control = control_resamples(save_pred = TRUE)
  )

tree_preds |>
  collect_predictions() |>
  roc_curve(truth = children, .pred_children)
## # A tibble: 22 × 3
##    .threshold specificity sensitivity
##         <dbl>       <dbl>       <dbl>
##  1   -Inf          0            1    
##  2      0.335      0            1    
##  3      0.338      0.0762       0.945
##  4      0.339      0.146        0.898
##  5      0.343      0.227        0.848
##  6      0.345      0.303        0.806
##  7      0.346      0.387        0.764
##  8      0.347      0.474        0.723
##  9      0.348      0.567        0.679
## 10      0.349      0.648        0.645
## # ℹ 12 more rows

Area under the curve

• AUC = 0.5: random guessing

• AUC = 1: perfect classifier

• In general AUC of above 0.8 considered “good”

⌨️ Visualize the ROC AUC

Instructions

Add an autoplot() to visualize the ROC AUC.

tree_preds |>
  collect_predictions() |>
  roc_curve(truth = children, .pred_children) |>
  autoplot()

tree_preds |>
  collect_predictions() |>
  roc_auc(truth = children, .pred_children)
## # A tibble: 1 × 3
##   .metric .estimator .estimate
##   <chr>   <chr>          <dbl>
## 1 roc_auc binary         0.670

Wrap up

Recap

• Machine learning is an extremely vast field containing a range of methodologies
• Supervised learning is used to predict outcomes
• To avoid bias in your model, use training/testing sets and cross-validation to partition your data
• Assess model performance using appropriate metrics on the assessment sets

Acknowledgments

• Materials derived from Tidymodels, Virtually by Allison Hill and licensed under a Creative Commons Attribution-ShareAlike 4.0 International (CC BY-SA) License.
• Dataset and some modeling steps derived from A predictive modeling case study and licensed under a Creative Commons Attribution-ShareAlike 4.0 International (CC BY-SA) License.