thresh_plot: Produce a precision-recall plot for a set of predicted...

In gweissman/gmish: Miscellaneous Functions for Predictive Modeling

Produce a precision-recall plot for a set of predicted probabilities for a binary classifier.

Description

Produce a precision-recall plot for a set of predicted probabilities for a binary classifier.

Usage

thresh_plot(
  preds,
  obs,
  metrics = c("ppv", "npv", "fscore"),
  max_intervals = 1000
)

Arguments

max_intervals	The maximum number of thresholds to evaluate. Default = 1000.
form	A formula where the left-hand side is the variable representing the observed outcome, 0 or 1, and the right-hand side represents the column names of the different model probabilities.
data	A data frame that contains at least two columns, one of which is the observed outcome and the others that are predicted probabilities.

Examples

library(ranger)
library(palmerpenguins)
pp <- penguins[complete.cases(penguins),]
m1 <- ranger(species == 'Adelie' ~ island + bill_length_mm + flipper_length_mm + body_mass_g + sex,
      data = pp, probability = TRUE)
p_obj <- predict(m1, data = pp)
results <- data.frame(preds_m1 = p_obj$predictions[,2],
                       obs = pp$species == 'Adelie')
thresh_plot(p_obj$predictions[,2], pp$species == 'Adelie')

gweissman/gmish documentation built on Sept. 3, 2024, 11:39 p.m.