plot_twoclass_pred_numcolR Documentation

Visualizing the predicted probabilities vs. a numeric column for a classification model

Description

This function plots the predicted probabilities against a numeric column based on tidymodels results for a two-class classification model.

Usage

plot_twoclass_pred_numcol(
  dat,
  y_name,
  numcol,
  alpha = 1,
  size = 1,
  prob_scaling = FALSE,
  event_level = "first",
  prob_eps = 0.001,
  source = NULL
)

Arguments

dat

The predictions data frame in the organize_data() result. Following variables are required: .outcome, .pred, .color, and .hover.

y_name

The y/response variable for the model.

numcol

The numerical column to plot against the predicted probabilities.

alpha

The opacity for the geom points.

size

The size for the geom points.

prob_scaling

The boolean to turn on or off the logit scale for probability.

event_level

A single character value for the level corresponding to the event.

prob_eps

A small numerical constant to prevent division by zero.

source

A character string of length 1 that matches the source argument in event_data().

Value

A plotly::ggplotly() object.