R/kfold_crossvalidation.R

In dsnavega/imputeForest: Missing value imputation model using randomForest

kfold_crossvalidation <- function(n, k = 10, seed = NULL) {

  # random number generator ----
  set.seed(seed = seed)

  # create folds and k-fold matrix ----
  x <- seq_len(n)
  kfold <- sample(cut(x = x, breaks = k, labels = FALSE))

  kfold_matrix <- matrix(NA, nrow = n, ncol = k)
  for(i in seq_len(k)) {
    kfold_matrix[, i] = kfold == i
  }

  colnames(kfold_matrix) <- seq_len(k)
  rownames(kfold_matrix) <- seq_len(n)

  # return ----
  rout <- kfold_matrix
  return(rout)

}