ROCR中的预测错误“预测和标签的交叉验证次数必须相等。”

Question

我正在尝试使用预测功能，但是有一个不允许我使用它。

pred <- predict(decision_tree, test_data, type = "prob") 

head(pred)
head(my_data)
tail(pred)
tail(test_data$Income)
pred <- prediction(pred, test_data$Income)
evalu <- performance(pred, "acc")

pred <- predict(decision_tree, test_data, type = "prob") 

head(pred)
head(my_data)
tail(pred)
tail(test_data$Income)
pred <- prediction(pred, test_data$Income)
evalu <- performance(pred, "acc")

我收到此错误：

Error in prediction(pred, test_data$Income) : 
  Number of cross-validation runs must be equal for predictions and labels.

Answer 1

嗨，我希望你在此期间工作。对于所有其他寻找答案的人：https://arulvelkumar.wordpress.com/2017/09/03/prediction-function-in-r-number-of-cross-validation-runs-must-be-equal-for-predictions-and-labels/ 简而言之：只需在您的行后添加一个 [,2] ： pred <- predict(decision_tree, test_data, type = "prob") [,2] 它应该可以工作。

问候 马蒂亚斯