设置 lgb.Dataset 对象的一个属性

来源： R/lgb.Dataset.R

设置 lgb.Dataset 的一个属性

set_field(dataset, field_name, data)

# S3 method for class 'lgb.Dataset'
set_field(dataset, field_name, data)

参数

dataset

类为 lgb.Dataset 的对象

field_name

要设置的属性名称字符串。可以是以下之一：

  • label: LightGBM 用于学习的标签；

  • weight: 用于进行权重缩放；

  • group: 用于排序学习任务。一个整数向量，描述如何将行分组为来自同一组候选结果的有序结果进行排序。例如，如果您有一个包含 100 个文档的数据集，且 group = c(10, 20, 40, 10, 10, 10)，这意味着您有 6 个组，其中前 10 条记录在第一个组中，记录 11-30 在第二个组中，依此类推。

  • init_score: 初始分数是 LightGBM 将从其基础上进行提升的起始预测。

data

字段的数据。请参见示例。

返回值

您传入的 lgb.Dataset 对象。

示例

# \donttest{
data(agaricus.train, package = "lightgbm")
train <- agaricus.train
dtrain <- lgb.Dataset(train$data, label = train$label)
lgb.Dataset.construct(dtrain)

labels <- lightgbm::get_field(dtrain, "label")
lightgbm::set_field(dtrain, "label", 1 - labels)

labels2 <- lightgbm::get_field(dtrain, "label")
stopifnot(all.equal(labels2, 1 - labels))
# }