If you want to use validation data, you should set reference to training data
lgb.Dataset.set.reference(dataset, reference)
dataset | object of class |
---|---|
reference | object of class |
passed dataset
library(lightgbm) data(agaricus.train, package ="lightgbm") train <- agaricus.train dtrain <- lgb.Dataset(train$data, label = train$label) data(agaricus.test, package = "lightgbm") test <- agaricus.test dtest <- lgb.Dataset(test$data, test = train$label) lgb.Dataset.set.reference(dtest, dtrain)