From dbe816a4dd8929045fdec5d9e4360e050191bb0b Mon Sep 17 00:00:00 2001 From: bjaeger Date: Sat, 7 Oct 2023 20:13:18 -0400 Subject: [PATCH] try collapse na_omit and drop the oobag guard --- R/orsf.R | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/R/orsf.R b/R/orsf.R index 2a15d303..c3607c23 100644 --- a/R/orsf.R +++ b/R/orsf.R @@ -461,12 +461,6 @@ orsf <- function(data, ) - # TODO: drop this; importance is computed with mort - if(importance %in% c("permute", "negate") && !oobag_pred){ - # oobag_pred <- TRUE # Should I add a warning? - oobag_pred_type <- 'surv' - } - if(is.null(oobag_fun)){ f_oobag_eval <- function(x) x @@ -651,9 +645,9 @@ orsf <- function(data, }, 'omit' = { - # data <- collapse::na_omit(data, cols = names_x_data) - cc <- stats::complete.cases(data[, names_x_data]) - data <- data[cc, ] + data <- collapse::na_omit(data, cols = names_x_data) + # cc <- stats::complete.cases(data[, names_x_data]) + # data <- data[cc, ] }, 'impute_meanmode' = {