R/xgb.DMatrix.R
slice.xgb.DMatrix.Rd
Get a new DMatrix containing the specified rows of original xgb.DMatrix object
slice(object, ...) # S3 method for xgb.DMatrix slice(object, idxset, ...) # S3 method for xgb.DMatrix [(object, idxset, colset = NULL)
object | Object of class "xgb.DMatrix" |
---|---|
... | other parameters (currently not used) |
idxset | a integer vector of indices of rows needed |
colset | currently not used (columns subsetting is not available) |
data(agaricus.train, package='xgboost') train <- agaricus.train dtrain <- xgb.DMatrix(train$data, label=train$label) dsub <- slice(dtrain, 1:42) labels1 <- getinfo(dsub, 'label') dsub <- dtrain[1:42, ] labels2 <- getinfo(dsub, 'label') all.equal(labels1, labels2)#> [1] TRUE