xgb.DMatrix 错误：标签的长度必须等于输入数据中的行数

Question

我在 R 中使用 xgboost。

我使用矩阵作为输入很好地创建了 xgb 矩阵，但是当我减少矩阵数据中的列数时，我收到错误。

这有效：

> dim(ctt1)

[1] 6401 5901

> xgbmat1 <- xgb.DMatrix(
     Matrix(data.matrix(ctt1)),
     label = as.matrix(as.numeric(data$V2)) - 1
  )

这不会：

> dim(ctt1[,nr])

[1] 6401 1048

xgbmat1 <- xgb.DMatrix(
    Matrix(data.matrix(ctt1[,nr])),
    label = as.matrix(as.numeric(data$V2)) - 1)

xgb.setinfo（dmat，names（p），p[[1]]）中的错误：标签的长度必须等于输入数据中的行数

Answer 1

就我而言，我通过更改分配操作修复了此错误：

labels <- df_train$target_feature