R混淆矩阵错误-分类树

Question

我是 R 的新手，如果这是一个简单的问题，请提前致歉。

我已经使用 rpart 包制作了一个决策树分类模型。

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

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

员工测试 table 包含 100 多个变量，但我感兴趣的字段如下所示：

summary(EmployeeTest$Leaver)

    Leaver Non-Leaver 
    66         86

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

提前致谢。

Answer 1

问题已解决：如果我更改

table_mat <- table(EmployeeTest$Leaver,pred)

至

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

尽管 pred 中只有一个字段 table

，但它是如何修复错误的

R confusion matrix error - classification tree