kerasformula (version 1.0.0)

confusion: confusion

Description

Confusion matrix or (for larger number of levels) confusion table.

Usage

confusion(object = NULL, y_test = NULL, predictions = NULL,
  return_xtab = NULL, digits = 3)

Arguments

object

Optional fit object. confusion() assumes object contains holdout/vaidation data as `y_test` and the forecasts/classifications as `predictions` but alternative variable names can be specified with the input arguments by those names.

y_test

A vector of holdout/validation data or the name in object (if fit object provided but alternative variable name required).

predictions

A vector predictions or the name in object (if fit object provided but alternative variable name required).

return_xtab

Logical. If TRUE, returns confusion matrix, which is a crosstable with correct predictions on the diagonal (if all levels are predicted at least once). If FALSE, returns (rectangular) table with columns for percent correct, most common misclassification, second most common misclassification, and other predictions. Defaults to TRUE (crosstable-style) only if number of levels < 6.

digits

Number of digits for proportions when return_xtab=FALSE; if NULL, no rounding is performed.

Value

confusion matrix or table as specified by return_xtab. #' @examples mtcars$make <- unlist(lapply(strsplit(rownames(mtcars), " "), function(tokens) tokens[1])) company <- if(is_keras_available()) kms(make ~ ., mtcars) else list(y_test = mtcars$make[1:5], predictions = sample(mtcars$make, 5)) confusion(company) # same as above confusion$company if is_keras_available() == TRUE confusion(company, return_xtab = FALSE) # focus on pCorrect, most common errors