R/createResample.R

In caret: Classification and Regression Training

#' @rdname createDataPartition
#' @export
createResample <- function(y, times = 10, list = TRUE) {
  if (inherits(y, "Surv"))
    y <- y[, "time"]
  trainIndex <- matrix(0, ncol = times, nrow = length(y))
  out <- apply(
    trainIndex, 2,
    function(data) {
      index <- seq(along = data)
      out <-
        sort(sample(index, size = length(index), replace = TRUE))
      out
    }
  )

  if (list)  {
    out <- as.data.frame(out, stringsAsFactors = TRUE)
    attributes(out) <- NULL
    names(out) <- prettySeq(out)
  } else {
    colnames(out) <- prettySeq(1:ncol(out))
  }

  out
}