https://github.com/berndbischl/mlr
Raw File
Tip revision: 3d7e0aa91936e82cf108aff3c46b19e3f953eefd authored by pat-s on 10 January 2020, 22:23:02 UTC
Bump version to 2.17.0.9000
Tip revision: 3d7e0aa
asROCRPrediction.R
#' Converts predictions to a format package ROCR can handle.
#'
#' @template arg_pred
#' @export
#' @family roc
#' @family predict
asROCRPrediction = function(pred) {
  UseMethod("asROCRPrediction")
}

asROCRPredictionIntern = function(probabilites, truth, negative, positive) {
  ROCR::prediction(probabilites, truth, label.ordering = c(negative, positive))
}

#' @export
asROCRPrediction.Prediction = function(pred) {
  if (length(pred$task.desc$class.levels) != 2L) {
    stop("More than 2 classes!")
  }
  asROCRPredictionIntern(getPredictionProbabilities(pred), pred$data$truth, pred$task.desc$negative, pred$task.desc$positive)
}

#' @export
asROCRPrediction.ResamplePrediction = function(pred) {
  if (length(pred$task.desc$class.levels) != 2L) {
    stop("More than 2 classes!")
  }
  prob = getPredictionProbabilities(pred)
  iter = pred$data$iter
  prob = split(prob, iter)
  truth = split(pred$data$truth, iter)
  ROCR::prediction(prob, truth, label.ordering = c(pred$task.desc$negative, pred$task.desc$positive))
}
back to top