Dimensions of an lgb.Dataset

Returns a vector of numbers of rows and of columns in an lgb.Dataset.

# S3 method for lgb.Dataset
dim(x, ...)

Arguments

x	Object of class lgb.Dataset
...	other parameters

Value

a vector of numbers of rows and of columns

Details

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

Examples

library(lightgbm)
data(agaricus.train, package = "lightgbm")
train <- agaricus.train
dtrain <- lgb.Dataset(train$data, label = train$label)

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