xgb_folds: Folds for xgboost

In bcjaeger/xgboost.surv: Extreme Gradient Boosting for Survival Analysis

Description

When calling xgb.cv, analysts may wish to manually specify folds rather than nfolds. This function generates an object that can be given as an input for folds and additionally lets the analyst specify stratifying variables to create balanced folds.

Usage

xgb_folds(data, nfolds, strata = NULL)

Arguments

data	a data.frame or tibble that will be used to generate folds.
nfolds	(integer) the number of folds
strata	A variable that is used to conduct stratified sampling to create the folds. This could be a single character value or a variable name that corresponds to a variable in data.

Examples

df <- data.frame(a=1:100, b=100:1, c=rep(letters[1:2],50))

xgb_folds(data = df, nfolds = 10)
xgb_folds(data = df, nfolds = 100)
xgb_folds(data = df, nfolds = 10, strata = c)

bcjaeger/xgboost.surv documentation built on Nov. 18, 2019, 6:43 a.m.