Formatted confusion matrix table

Produces a styled gt confusion matrix with correct predictions highlighted. Only available for classification models.

Usage

tl_table_confusion(model, new_data = NULL, ...)

Arguments

model

A tidylearn classification model

new_data

Optional test data. If NULL, uses training data.

...

Additional arguments (currently unused)

Value

A gt table object, with a row and a column for each class the model was trained on. Rows of a class the model never saw are left out, with a warning.

Examples

# \donttest{
model <- tl_model(iris, Species ~ ., method = "forest")
tl_table_confusion(model)


  
Confusion Matrix
    
Actual
      

        
Predicted
      
    
setosa
      
versicolor
      
virginica
    
setosa
50
0
0
versicolor
0
50
0
virginica
0
0
50
tidylearn | forest (classification) | Species ~ . | n = 150
    
# }