lgb.slice.Dataset function

Slice a dataset

Get a new lgb.Dataset containing the specified rows of original lgb.Dataset object

Renamed from slice() in 4.4.0

lgb.slice.Dataset(dataset, idxset)

Arguments

• dataset: Object of class lgb.Dataset
• idxset: an integer vector of indices of rows needed

Returns

constructed sub dataset

Examples

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

dsub <- lgb.slice.Dataset(dtrain, seq_len(42L))
lgb.Dataset.construct(dsub)
labels <- lightgbm::get_field(dsub, "label")