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 of length \(n_1,\ldots,n_q\)

family

character "gaussian" or "binomial"

nfolds

integer between 2 and \(n\)

Value

Returns the fold identifiers in a vector of length \(n\) with entries 1,\(\ldots\),nfolds (multi-task learning) on in a list of \(q\) vectors of lengths \(n_1,\ldots,n_q\) (transfer learning).

Examples

#--- multi-task learning ---
family <- "binomial"
y <- sim_data_multi(family=family)$y_train
fold <- make_folds_multi(y=y,family=family)

#--- transfer learning ---
family <- "binomial"
y <- sim_data_trans(family=family)$y_train
fold <- make_folds_trans(y,family=family)