R/measure_kappa.R

Defines functions measure_kappa

Documented in measure_kappa

#' Cohen's kappa
#'
#' This function is based on function `measureKAPPA` from `mlr` package.
#'
#' @param truth a vector with true (reference) values.
#' @param response a vector with response (predicted) values.
#' @param conf_mat a table similar to (\code{table(truth, response, useNA = "no")}).
#'
#' @return [!!!]
#'
#' @export
#' @family measures_
#'
#' @examples
#'
#' truth <- rep(1:3, times = 50)
#' prediction <- rep(3:1, each = 50)
#'
#' measure_kappa(truth, prediction)
#'
#' square_matrix <- table(truth, prediction)
#' measure_kappa(conf_mat = square_matrix)
#'
#'
#' # # matrix(c("TP", "FN", "FP", "TN"), 2)
#' #
#' # make_conf_matrix <- function(TP, FN, FP, TN)
#' #                  matrix(c(TP, FN, FP, TN), 2)
#' #
measure_kappa <- function(truth = NULL, response = NULL, conf_mat = NULL) {
  if (is.null(conf_mat)) {
    conf_mat <- table(truth, response)
  }
  if (nrow(conf_mat) != ncol(conf_mat)) {
    stop("Confusion matrix `conf_mat` must be square.")
  }

  conf_mat <- conf_mat / sum(conf_mat)
  # p observed:
  p0 <- sum(diag(conf_mat))
  # p expectrd:
  rowsum <- rowSums(conf_mat)
  colsum <- colSums(conf_mat)
  pe <- sum(rowsum * colsum) / sum(conf_mat)^2

  # Cohen's kappa:
  (1 - (1 - p0) / (1 - pe))
}
GegznaV/multiROC documentation built on Sept. 15, 2020, 10:33 a.m.