make_folds_multi {sparselink}R Documentation

Create folds for multi-task and transfer learning

Description

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)


[Package sparselink version 1.0.0 Index]