viz_class_fit: Visualize a classification model output

In snfagora/autotextclassifier: R Package for Automatically Classifying Texts Based on Tidymodels

Visualize a classification model output

Description

Visualize a classification model output

Usage

viz_class_fit(
  model,
  model_title,
  test_x_class,
  test_y_class,
  metric_type = "class"
)

Arguments

model	A classification model output
model_title	A classification model name
test_x_class	Outcomes of the test dataset
test_y_class	Predictors of the test dataset
metric_type	A type of the metrics ("class" or "probability"). Either a "class"-based metrics (e.g., accuracy, balanced accuracy, F-score) or a "probability"-based metrics (e.g., ROC Curve). The default type is "class."

Value

a bar or line plot(s)

snfagora/autotextclassifier documentation built on June 16, 2022, 9:42 a.m.