xgboost (version 1.7.6.1)

dim.xgb.DMatrix: Dimensions of xgb.DMatrix

Description

Returns a vector of numbers of rows and of columns in an xgb.DMatrix.

Usage

# S3 method for xgb.DMatrix
dim(x)

Arguments

x

Object of class xgb.DMatrix

Details

Note: since nrow and ncol internally use dim, they can also be directly used with an xgb.DMatrix object.

Examples

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

stopifnot(nrow(dtrain) == nrow(train$data))
stopifnot(ncol(dtrain) == ncol(train$data))
stopifnot(all(dim(dtrain) == dim(train$data)))