Learn R Programming

xgboost (version 1.7.6.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

Run this code
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))

Run the code above in your browser using DataLab