Skip to contents

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 n1,,nq

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,,nfolds (multi-task learning) on in a list of q vectors of lengths n1,,nq (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)