Ошибка в `$ <-. Data.frame` (` * tmp * `, прогноз, значение = c (` 1` = 1L, `2` = 1L,: замена имеет 3462 строки, данные имеют 1538

Question

Я получаю вышеуказанную ошибку в R, когда пытаюсь предсказать и оценить дерево сокращенных поездов на наборе данных тестирования

> dim(test)
[1] 1538   13

> test$prediction = predict(ptrain_tree, data = test, type="class")

**Error in `$<-.data.frame`(`*tmp*`, prediction, value = c(`1` = 1L, `2` = 1L,  : 
  replacement has 3462 rows, data has 1538**

> test$prob1 = predict(ptrain_tree, data = test, type = "prob" )[,"1"]

**Error in `$<-.data.frame`(`*tmp*`, prob1, value = c(`1` = 0.004296875,  : 
  replacement has 3462 rows, data has 1538**

ПРИМЕЧАНИЕ. - ptrain_tree - это мое сокращенное дерево, созданное на наборе данных поездов.

Answer 1

Функция прогнозирования для тестового набора данных немного хитрая. Это работает, только если вы пропустите data=.

Попробуйте: test$prediction = predict(ptrain_tree, test, type="class")