R/get.max_sensitivity.R

In Laurae2/Laurae: Advanced High Performance Data Science Toolbox for R

#' Maximum Sensitivity (True Positive Rate)
#' 
#' This function allows to use a custom thresholding method to maximize the Sensitivity (True Positive Rate). A pair of values is returned.
#' 
#' @param preds Type: numeric. The predictions.
#' @param labels Type: numeric. The labels (0, 1).
#' 
#' @return A two element vector containing the maximum Sensitivity (True Positive Rate) for binary data, and the threshold used.
#' 
#' @export

get.max_sensitivity <- function(preds, labels) {
  
  DT <- data.table(y_true = labels, y_prob = preds, key = "y_prob")
  cleaner <- !duplicated(DT[, "y_prob"], fromLast = TRUE)
  nump <- sum(labels)
  numn <- length(labels) - nump
  
  DT[, fn_v := numn - as.numeric(cumsum(y_true == 0))]
  DT[, tp_v := nump - cumsum(y_true == 1)]
  DT <- DT[cleaner, ]
  DT[, sens := tp_v / (tp_v + fn_v)]
  
  best_row <- which.max(DT$sens)
  
  if (length(best_row) > 0) {
    return(c(DT$sens[best_row[1]], DT$y_prob[best_row[1]]))
  } else {
    return(c(-1, -1))
  }
  
}