R混淆矩阵错误-分类树

Question

我对R很陌生，所以如果这是一个简单的问题，请提前道歉。

我已经使用rpart软件包建立了决策树分类模型。

这很好用，我用它来预测测试数据中的变量。我正在尝试制作一个混淆矩阵表来比较结果，但是我不断收到以下错误：

Error in table(EmployeeTest$Leaver, pred) : 
all arguments must have the same length

我已经上网了，似乎找不到解决方法。

我用于创建矩阵的代码如下：

table_mat <- table(EmployeeTest$Leaver,pred)

pred是模型的预测结果，如下所示：

summary(pred)

         pred   
 Leaver    :61  
 Non-Leaver:91 

“员工测试”表包含100多个变量，但我感兴趣的字段如下：

summary(EmployeeTest$Leaver)

    Leaver Non-Leaver 
    66         86

由于我不知道如何解决此问题，因此我们将不胜感激任何帮助。我知道错误提示这与长度有关，但是两个表都包含具有相同长度的相同变量。

谢谢。

Answer 1

问题已解决：如果我更改了

table_mat <- table(EmployeeTest$Leaver,pred)

到

table_mat <- table(EmployeeTest$Leaver,pred$pred)

尽管在pred表中只有一个字段，但该错误的修复方法