r - 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 

Employee Test 表包含 100 多个变量，但我感兴趣的字段如下所示：

summary(EmployeeTest$Leaver)

    Leaver Non-Leaver 
    66         86

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

提前致谢。

标签： rconfusion-matrix

Answer 1

解决的问题：如果我改变了

table_mat <- table(EmployeeTest$Leaver,pred)

到

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

尽管 pred 表中只有一个字段，但它如何修复了错误