缺少类别的 R 包插入符号混淆矩阵

Question

我正在使用功能 confusionMatrix在 R 包裹caret为我拥有的一些数据计算一些统计数据。我一直在将我的预测以及我的实际值放入 table获取要在 confusionMatrix 中使用的表的函数功能如下:

table(predicted,actual)

然而，有多种可能的结果(例如 A、B、C、D)，我的预测并不总是代表所有的可能性(例如只有 A、B、D)。 table 的结果输出函数不包括缺失的结果，如下所示:

    A    B    C    D
A  n1   n2   n2   n4  
B  n5   n6   n7   n8  
D  n9  n10  n11  n12
# Note how there is no corresponding row for `C`.

confusionMatrix函数无法处理丢失的结果并给出错误:

Error in !all.equal(nrow(data), ncol(data)) : invalid argument type

有什么方法可以使用 table以不同的方式获取带有零的缺失行或使用 confusionMatrix功能不同，所以它会将丢失的结果视为零？

请注意:由于我是随机选择要测试的数据，因此有时实际结果中也没有表示类别，而只是预测结果。我不相信这会改变解决方案。

Answer 1

您可以使用 union以确保相似的水平:

library(caret)

# Sample Data
predicted <- c(1,2,1,2,1,2,1,2,3,4,3,4,6,5) # Levels 1,2,3,4,5,6
reference <- c(1,2,1,2,1,2,1,2,1,2,1,3,3,4) # Levels 1,2,3,4

u <- union(predicted, reference)
t <- table(factor(predicted, u), factor(reference, u))
confusionMatrix(t)