Visualize a classification model output

Visualize a classification model output

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)