Set information of an xgb.DMatrix object

Description

Set information of an xgb.DMatrix object

Usage

setinfo(object, ...)

## S4 method for signature 'xgb.DMatrix'
setinfo(object, name, info)

Arguments

Details

It can be one of the following:

• label: label Xgboost learn from ;

• weight: to do a weight rescale ;

• base_margin: base margin is the base prediction Xgboost will boost from ;

• group.

Examples

data(agaricus.train, package='xgboost')
train <- agaricus.train
dtrain <- xgb.DMatrix(train$data, label=train$label)
labels <- getinfo(dtrain, 'label')
setinfo(dtrain, 'label', 1-labels)
labels2 <- getinfo(dtrain, 'label')
stopifnot(all(labels2 == 1-labels))

Results