R/predict_DT.R

In CedricMondy/ecodiag: Build Standardized Diagnostic Tool Models

#' @import mlr
predict_DT <- function(object,
                       newdata,
                       pred.all = TRUE) {

  funScale <- approxfun(x = c(0, object$threshold, 1),
                        y = c(0, 0.5, 1))

  IP_all <- pbapply::pblapply(object$models,
                              mlr:::predict.WrappedModel,
                              newdata = newdata) %>%
    lapply(function(pred) {
      funScale(pred$data$prob.impaired)
    })                                           %>%
    do.call(what = cbind)                        %>%
    data.frame()                                 %>%
    dplyr::as.tbl()
  colnames(IP_all) <- paste("iter", 1:length(object$models), sep = "_")


  IP_summ <- apply(X = IP_all, MARGIN = 1, FUN = mean) %>%
    data.frame()                    %>%
    dplyr::as.tbl()
  colnames(IP_summ) <- "average"

  if (pred.all) {
    return(list(IP_all = IP_all, IP_summ = IP_summ))
  } else {
    return(IP_summ)
  }
}