R/collapse_misc.R

In bcjaeger/orsf2: Accelerated Oblique Random Forests

as_matrix <- function(x){
 collapse::qM(x)
}

#' subset data by columns
#'
#' Made this function so that data.table inputs wouldn't require
#'   special treatment
#'
#' @param data data to select columns from
#' @param col_names names of the columns to select
#'
#' @return data[, col_names]
#' @noRd
#'
select_cols <- function(data, col_names){

 collapse::fselect(data, as.character(col_names), return = 'data')

}


data_impute <- function(data, cols, values){

 for(col in cols){

  if(anyNA(data[[col]])){

   value <- values[[col]]

   # coerce mean to integer if col is integer
   if(is.integer(data[[col]])){
    value <- as.integer(value)
   }

   data <- collapse::replace_NA(data,
                                cols = col,
                                value = value)

  }


 }

 data

}


# data_impute_nocheck <- function(data, cols, values){
#
#  for(col in cols)
#   data <- collapse::replace_NA(data,
#                                cols = col,
#                                value = values[[col]])
#
#  data
#
# }