foldgen: Stratified cross-validation fold generator function, inspired...

In nlmixr2extra: Nonlinear Mixed Effects Models in Population PK/PD, Extra Support Functions

Stratified cross-validation fold generator function, inspired from the caret

Description

Stratified cross-validation fold generator function, inspired from the caret

Usage

foldgen(data, nfold = 5, stratVar = NULL)

Arguments

data	data frame used in the analysis
nfold	number of k-fold cross validations. Default is 5
stratVar	Stratification Variable. Default is NULL and ID is used for CV

Value

return data.frame with the fold column attached

Author(s)

Vishal Sarsani, caret

Examples

d <- nlmixr2data::theo_sd
d$SEX <-0
d$SEX[d$ID<=6] <-1

covarsVec <- c("WT")

# Stratified cross-validation data with CMT
df1 <- foldgen(d, nfold=5, stratVar="CMT")

# Stratified cross-validation data with ID (individual)
df2 <- foldgen(d, nfold=5, stratVar=NULL)

nlmixr2extra documentation built on April 12, 2025, 1:41 a.m.