produceFolds: Produce folds

In ecpc: Flexible Co-Data Learning for High-Dimensional Prediction

Produce folds

Description

Produce folds for cross-validation.

Usage

produceFolds(nsam, outerfold, response, model = c("logistic","cox","other"), 
balance = TRUE)

Arguments

nsam	Number of samples
outerfold	Number of folds.
response	Response data.
model	Type of model for the response.
balance	Should folds be balanced in response?

Value

A list with ‘outerfold’ elements containing a vector of sample indices in each fold.

Examples

n<-100
outerfold <- 10

#linear model
resp <- rnorm(n)
folds <- produceFolds(nsam=n, outerfold=outerfold, response=resp)

#logistic model: keep 0/1 balanced across folds
resp <- as.factor(rnorm(n)>0.5)
folds <- produceFolds(nsam=n, outerfold=outerfold, response=resp, balance = TRUE)

ecpc documentation built on March 7, 2023, 6:46 p.m.