Stratified CV to insure balance (by one grouping variable, Y)

Creates a dummy variable that partitions the data into v equal sized groups for v-fold CV.

create_cv_folds(V, Y, verbose = F)

Arguments

V

number of folds

Y

Outcome variable. If binary will be used for stratification.

verbose

If T will display extra output.

Value

Vector of fold assignments.