R/rfsrc_pred.R

In Mthrun/dbt.FlowCytometry: Databionics Toolbox for the Analysis of Fluorescence-Activated Cell Sorting (FACS) Data

predict_model.rfsrc <- function(x, newdata, type, ...) {
  res <- predict(x, newdata = newdata, ...)
  x=res$predicted
  switch(
    type,
    raw = data.frame(Response = apply(predicted, 1, which.max), stringsAsFactors = FALSE),
    prob = as.data.frame(x, check.names = FALSE)
  )
}
model_type.rfsrc<- function(x, ...) 'classification'