generatefolds: Generate random folds for cross-validation

View source: R/generatefolds.R

Generate random folds for cross-validationR Documentation

Generate random folds for cross-validation

Description

Random folds for use in a cross validation are generated. There is the option for stratified splitting as well.

Usage

generatefolds(target, nfolds = 10, stratified = TRUE, seed = FALSE) 

Arguments

target

A vector with some data, either continuous or categorical.

nfolds

The number of folds to produce.

stratified

A boolean variable specifying whether stratified random (TRUE) or simple random (FALSE) sampling is to be used when producing the folds.

seed

A boolean variable. If set to TRUE, the folds will always be the same.

Details

I was inspired by the sam command in the package TunePareto in order to do the stratified version.

Value

A list with nfolds elements where each elements is a fold containing the indices of the data.

Author(s)

Michail Tsagris

R implementation and documentation: Michail Tsagris mtsagris@uoc.gr

See Also

cv.ses

Examples

a <- generatefolds(iris[, 5], nfolds = 5, stratified = TRUE)
table(iris[a[[1]], 5])  ## 10 values from each group

MXM documentation built on Aug. 25, 2022, 9:05 a.m.