时间戳

首页 > 解决方案 > xgboost 在 R 中设置

r - xgboost 在 R 中设置

问题描述

我正在尝试在 R 中使用 xgboost，但我不确定模型的输入需要是什么。我的模型代码是：

xgb <- xgboost(data = data.matrix(training[,-1]), 
                           label = labels, 
                           eta = 0.1,
                           max_depth = 15, 
                           nround=25, 
                           subsample = 0.5,
                           colsample_bytree = 0.5,
                           seed = 1,
                           eval_metric = "merror",
                           objective = "binary:logistic",
                           num_class = 12,
                           nthread = 3
            )

我收到以下错误：

setinfo.xgb.DMatrix(dmat, names(p), p[[1]]) 中的错误：标签的长度必须等于输入数据中的行数

我的标签是我要预测的列，但输入数据是所有输入变量。例如，标签的暗淡显示 [1] 1837 1 和训练表的暗淡是 [1] 1837 8 所以我认为这些不必是相同的大小？标签列设置为真或假，因为我认为这是所需的格式。具有真或假的列是训练数据中的最后一列，因此我使用训练维度，[-1]。

有人可以解释一下其余的输入是什么，因为我从网上的一个例子中复制了这段代码，我不明白需要什么。

标签： rxgboost

解决方案

推荐阅读