conf_mat_resampled: Compute average confusion matrix across resamples

In tune: Tidy Tuning Tools

Compute average confusion matrix across resamples

Description

For classification problems, conf_mat_resampled() computes a separate confusion matrix for each resample then averages the cell counts.

Usage

conf_mat_resampled(x, ..., parameters = NULL, tidy = TRUE)

Arguments

x	An object with class tune_results that was used with a classification model that was run with control_*(save_pred = TRUE).
...	Currently unused, must be empty.
parameters	A tibble with a single tuning parameter combination. Only one tuning parameter combination (if any were used) is allowed here.
tidy	Should the results come back in a tibble (TRUE) or a conf_mat object like yardstick::conf_mat() (FALSE)?

Value

A tibble or conf_mat with the average cell count across resamples.

Examples

library(parsnip)
library(rsample)
library(dplyr)

data(two_class_dat, package = "modeldata")

set.seed(2393)
res <-
  logistic_reg() %>%
  set_engine("glm") %>%
  fit_resamples(
    Class ~ .,
    resamples = vfold_cv(two_class_dat, v = 3),
    control = control_resamples(save_pred = TRUE)
  )

conf_mat_resampled(res)
conf_mat_resampled(res, tidy = FALSE)

tune documentation built on May 29, 2024, 7:32 a.m.