plot_twoclass_pred_factorcol: Visualizing the predicted probabilities vs. a factor variable...

View source: R/two_class_diag_plots.R

plot_twoclass_pred_factorcolR Documentation

Visualizing the predicted probabilities vs. a factor variable for a classification model

Description

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

Usage

plot_twoclass_pred_factorcol(
  dat,
  y_name,
  factorcol,
  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.

factorcol

The factor 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.


shinymodels documentation built on May 29, 2024, 2:22 a.m.