R/getPerformanceMeasuresForSingleRankAsDataFrame.R

In felipepratalima/gunga:

getPerformanceMeasureForSingleRankAsDataFrame <- function(expectedIds, predictedIds) {
  expectedIds <- unique(expectedIds)
  predictedIds <- unique(predictedIds)

  currentTP <- sum(predictedIds %in% expectedIds)
  currentFP <- sum(predictedIds %in% expectedIds == F)
  currentTN <- 0
  currentExpectedTP <- length(expectedIds)
  currentFN <- abs(currentExpectedTP  - currentTP)
  currentPrecision <- currentTP / (currentTP + currentFP)
  currentRecall <- currentTP / (currentTP + currentFN)
  currentF1 <- (2 * currentPrecision * currentRecall) / (currentPrecision + currentRecall)
  currentAccuracy <- (currentTP + currentTN) / (currentTP + currentFP + currentFN + currentTN)
  currentSpecificity <- currentTN / (currentTN + currentFP)

  performanceDf <- data.frame(
    expectedTP = currentExpectedTP,
    TP = currentTP,
    FP = currentFP,
    FN = currentFN,
    TN = currentTN,
    Precision = currentPrecision,
    Recall = currentRecall,
    F1 = currentF1,
    Accuracy = currentAccuracy,
    Specificity = currentSpecificity
  )
  performanceDf
}