create.stratified.fold.df: DataFrame for stratified cross validation

In HEMDAG: Hierarchical Ensemble Methods for Directed Acyclic Graphs

Description

Create a data frame for stratified cross-validation.

Usage

create.stratified.fold.df(labels, scores, folds = 5, seed = 23)

Arguments

labels	vector of the true labels (0 negative, 1 positive).
scores	a numeric vector of the values of the predicted labels.
folds	number of folds of the cross validation (def. folds=5).
seed	initialization seed for the random generator to create folds (def. seed=23). If seed=NULL, the stratified folds are generated without seed initialization.

Details

Folds are stratified, i.e. contain the same amount of positive and negative examples.

Value

A data frame with three columns:

• scores: contains the predicted scores;

• labels: contains the labels as pos or neg;

• folds: contains the index of the fold in which the example falls. The index can range from 1 to the number of folds.

Examples

data(labels);
data(scores);
df <- create.stratified.fold.df(L[,3], S[,3], folds=5, seed=23);

HEMDAG documentation built on Feb. 12, 2021, 5:13 p.m.