获取 lgb.Dataset 的一个属性

get_field(dataset, field_name)

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

参数

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 将从中提升的基础预测。

返回值

请求的属性

示例

# \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(labels2 == 1 - labels))
# }