R/cv_indexation.mode_distribute.R

In DMRnet: Delete or Merge Regressors Algorithms for Linear and Logistic Model Selection and High-Dimensional Data

cv_indexation.mode_distribute <- function(X, y, nfolds, indexation.mode, model_function, ...) {

       X <- data.frame(X, check.names = TRUE, stringsAsFactors = TRUE)

       if (indexation.mode == "dimension") {
               out <- cv_MD_indexed(X, y, nfolds, model_function, ...)
       } else {
               if (indexation.mode == "GIC") {
                       out <- cv_GIC_indexed(X, y, nfolds, model_function, ...)
               } else {
                       stop("Error: wrong indexation.mode, should be one of: GIC, dimension")
               }
       }

       out$indexation.mode <- indexation.mode
       out$call <- match.call()
       class(out) <- "cv.DMR"

       return(out)
}