xgboost (version 1.6.0.1)

setinfo: Set information of an xgb.DMatrix object

Description

Set information of an xgb.DMatrix object

Usage

setinfo(object, ...)
# S3 method for xgb.DMatrix
setinfo(object, name, info, ...)

Arguments

object

Object of class "xgb.DMatrix"

...

other parameters

name

the name of the field to get

info

the specific field of information to set

Details

The name field 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: number of rows in each group (to use with rank:pairwise objective).

Examples

data(agaricus.train, package='xgboost')
dtrain <- with(agaricus.train, xgb.DMatrix(data, label = label))

labels <- getinfo(dtrain, 'label')
setinfo(dtrain, 'label', 1-labels)
labels2 <- getinfo(dtrain, 'label')
stopifnot(all.equal(labels2, 1-labels))