Create folds for multi-task and transfer learning

Create folds for multi-task and transfer learning

Usage

make.folds.multi(y, family, nfolds = 10)

make.folds.trans(y, family, nfolds = 10)

Arguments

y

multi-task learning: y matrix with \(n\) rows (samples) and \(q\) columns (outcomes) transfer learning: list of \(q\) numeric vectors

family

character "gaussian" or "binomial"

nfolds

integer between 2 and \(n\)

Examples

#--- multi-task learning ---
family <- "binomial"
y <- sim.data.multiple(family=family)$y_train
fold <- make.folds.multi(y=y,family=family)

#--- transfer learning ---
family <- "binomial"
y <- sim.data.transfer(family=family)$y_train
fold <- make.folds.trans(y,family=family)