r - xgb.DMatrix 错误 : The length of labels must equal to the number of rows in the input data

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)

Error in xgb.setinfo(dmat, names(p), p[[1]]) : The length of labels must equal to the number of rows in the input data

Answer 1

就我而言，我通过更改分配操作来修复此错误:
labels <- df_train$target_feature