[R] k-fold cv using caret package

k_fold_cv = function(data,  
                     .y_name,  
                     .repeated = TRUE,  
                     .repeated_times = 5,  
                     .seed_number = 777)  
{  
   library(caret)  
   library(data.table)

   set.seed(.seed_number)  
   y = unlist(data[, y_name, with = FALSE])

   if(repeated){  
      cv_splits = caret::createMultiFolds(y, k = .num_folds, times = .repeat_times)
   } else{  
      cv_splits = caret::createMultiFolds(y, k = .num_folds, returnTrain = TRUE)  
   }
   return(cv_splits)  
}