Многоклассовая классификация XGBOOST R
Помогите пожалуйста, возникла ошибка. Не могу разобраться почему. В наборе 13 столбцов. Всего 5 классов.
Error in setinfo.xgb.DMatrix(dmat, names(p), p[[1]]) : The length of labels must equal to the number of rows in the input data
train <- read.csv('D:/Class.csv')
train=as.data.frame(train)
trainObs <- sample(nrow(train), .8 * nrow(train), replace = FALSE)
valObs <- sample(nrow(train), .2 * nrow(train), replace = FALSE)
train_dat <- train[trainObs,]
val_dat <- train[valObs,]
train_labs <- as.numeric(train_dat$employmentstatus) - 1
val_labs <- as.numeric(val_dat$employmentstatus) - 1
new_train <- model.matrix(~ . + 0, data = train_dat[, -5])
new_val <- model.matrix(~ . + 0, data = train[valObs, -5])
xgb_train <- xgb.DMatrix(data = new_train, label = train_labs)
xgb_val <- xgb.DMatrix(data = new_val, label = val_labs)