r语言 - confusionMatrix函数和不同大小的矩阵

Question

我正在使用插入符号包使用confusionMatrix函数。该函数在2x2(或5x5等)矩阵中工作得很好。例子:

Table1:

   1  2
1 25 15
2  8 33
confusionMatrix(Table1) -> works!

但是如果我有一个不同大小的表:

表:

    5   6   7
3   1   1   0
4   8   5   0
5 153  57   3
6  57 105  19
7   6  27  27
8   0   3   6
confusionMatrix(Table2) -> Won't work!
Error: Error in !all.equal(nrow(data), ncol(data)) : invalid argument type

我怎么能绕过这个麻烦，因为我被迫使用confusionMatrix函数?

Answer 1

试试这个:

x <- as.integer(Prediction)
y <- Test$quality
l <- union(x, y)
Table2 <- table(factor(x, l), factor(y, l))
confusionMatrix(Table2)

我们的想法是将您对table的输入转换为两个共享完全相同级别(l)的因子。那么你可以保证Table2将是正方形的