1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
#' 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))
}
|
