setinfo: Set information of an xgb.DMatrix object

In xgboost: Extreme Gradient Boosting

Set information of an xgb.DMatrix object

Description

Set information of an xgb.DMatrix object

Usage

setinfo(object, ...)

## S3 method for class '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, nthread = 2))

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

xgboost documentation built on Sept. 11, 2024, 8:26 p.m.