设置一个 lgb.Dataset 对象的分类特征。使用此函数告诉 LightGBM 哪些特征应被视为分类特征。

lgb.Dataset.set.categorical(dataset, categorical_feature)

参数

dataset

lgb.Dataset 对象

categorical_feature

分类特征。这可以是特征名称的字符向量,或者是包含特征索引的整数向量(例如 c(1L, 10L),表示“第一列和第十列”)。

返回值

您传入的数据集

示例

# \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)
# }