设置一个 lgb.Dataset
对象的分类特征。使用此函数告诉 LightGBM 哪些特征应被视为分类特征。
lgb.Dataset.set.categorical(dataset, categorical_feature)
您传入的数据集
# \donttest{
data(agaricus.train, package = "lightgbm")
train <- agaricus.train
dtrain <- lgb.Dataset(train$data, label = train$label)
data_file <- tempfile(fileext = ".data")
lgb.Dataset.save(dtrain, data_file)
#> [LightGBM] [Info] Saving data to binary file /tmp/RtmpViPAUB/file13c56441b443.data
dtrain <- lgb.Dataset(data_file)
lgb.Dataset.set.categorical(dtrain, 1L:2L)
# }