R confusionMatrix错误数据和参考因素具有相同的水平

R confusionMatrix error data and reference factors with same levels

我试图了解在使用 glm 函数进行逻辑回归后如何制作混淆矩阵。到目前为止,这是我的代码。我正在使用 caret 包和 confusionMatrix 函数。

dput(head(wine_quality))

structure(list(fixed.acidity = c(7, 6.3, 8.1, 7.2, 7.2, 8.1), 
    volatile.acidity = c(0.27, 0.3, 0.28, 0.23, 0.23, 0.28), 
    citric.acid = c(0.36, 0.34, 0.4, 0.32, 0.32, 0.4), residual.sugar = c(20.7, 
    1.6, 6.9, 8.5, 8.5, 6.9), chlorides = c(0.045, 0.049, 0.05, 
    0.058, 0.058, 0.05), free.sulfur.dioxide = c(45, 14, 30, 
    47, 47, 30), total.sulfur.dioxide = c(170, 132, 97, 186, 
    186, 97), density = c(1.001, 0.994, 0.9951, 0.9956, 0.9956, 
    0.9951), pH = c(3, 3.3, 3.26, 3.19, 3.19, 3.26), sulphates = c(0.45, 
    0.49, 0.44, 0.4, 0.4, 0.44), alcohol = c(8.8, 9.5, 10.1, 
    9.9, 9.9, 10.1), quality = structure(c(4L, 4L, 4L, 4L, 4L, 
    4L), .Label = c("3", "4", "5", "6", "7", "8", "9", "white"
    ), class = "factor"), type = structure(c(3L, 3L, 3L, 3L, 
    3L, 3L), .Label = c("", "red", "white"), class = "factor"), 
    numeric_type = c(0, 0, 0, 0, 0, 0)), row.names = c(NA, 6L
), class = "data.frame")

library(tibble) 
library(broom) 
library(ggplot2)
library(caret)

any(is.na(wine_quality)) # this evaulates to FALSE 


wine_model <- glm(type ~ fixed.acidity + volatile.acidity + citric.acid + residual.sugar +  chlorides + free.sulfur.dioxide + total.sulfur.dioxide + density + pH + sulphates + alcohol, wine_quality, family = "binomial")


# split data into test and train

smp_size <- floor(0.75 * nrow(wine_quality))

set.seed(123)
train_ind <- sample(seq_len(nrow(wine_quality)), size = smp_size)

train <- wine_quality[train_ind, ]
test <- wine_quality[-train_ind, ]


# make prediction on train data

pred <- predict(wine_model)

train$fixed.acidity <- as.numeric(train$fixed.acidity)
round(train$fixed.acidity)
train$fixed.acidity <- as.factor(train$fixed.acidity)

pred <- as.numeric(pred)
round(pred)
pred <- as.factor(pred)

confusionMatrix(pred, wine_quality$fixed.acidity)

在这最后一行代码之后,我得到了这个错误:

Error: `data` and `reference` should be factors with the same levels.

这个错误对我来说没有意义。我已经测试过 pred 的长度和 fixed.acidity 的长度都是相同的 (6497),而且它们都是因子数据类型。

length(pred)
length(wine_quality$fixed.acidity)

class(pred)
class(train$fixed.acidity)

这个混淆矩阵不起作用有什么明显的原因吗?我正在尝试找到模型的命中率。我将不胜感激虚拟解释我真的不知道我在这里做什么。

来自confusionMatrix()的错误告诉我们传递给函数的两个变量需要是具有相同值的因子。当我们对两个变量 运行 str() 时,我们可以看到为什么会收到错误。

> str(pred)
 Factor w/ 5318 levels "-23.6495182533792",..: 310 339 419 1105 310 353 1062 942 594 1272 ...
> str(wine_quality$fixed.acidity)
 num [1:6497] 7.4 7.8 7.8 11.2 7.4 7.4 7.9 7.3 7.8 7.5 ...

pred 是一个因子,而 wine_quality$fixed_acidity 是一个数值向量。 confusionMatrix() 函数用于比较因变量的预测值和实际值。它无意交叉制表预测变量和自变量。

问题中的代码在混淆矩阵中使用了 fixed.acidity,而它应该将 type 的预测值与来自测试数据的 type 的实际值进行比较。

此外,问题中的代码在将数据拆分为测试数据和训练数据之前创建了模型。正确的程序是在训练数据上建立模型之前拆分数据,用测试(阻止)数据进行预测,并将实际情况与测试数据中的预测进行比较。

最后,原始post中编码的predict()函数的结果是GLM模型的线性预测值(相当于输出模型对象中的wine_model$linear.predictors) .在 confusionMatrix().

中使用之前,必须进一步转换这些值以使其适用

实际上,将 caret::train() 与 GLM 方法和二项式族一起使用会更容易,其中 predict() 将生成可用于 confusionMatrix() 的结果。我们将用 UCI 葡萄酒质量数据来说明这一点。

首先,我们从 UCI 机器学习存储库下载数据,使示例可重现。

download.file("https://archive.ics.uci.edu/ml/machine-learning-databases/wine-quality/winequality-red.csv",
              "./data/wine_quality_red.csv")
download.file("https://archive.ics.uci.edu/ml/machine-learning-databases/wine-quality/winequality-white.csv",
              "./data/wine_quality_white.csv")

其次,我们加载数据,根据数据文件将 type 分配为红色或白色,并将数据绑定到单个数据框中。

red <- read.csv("./data/wine_quality_red.csv",header = TRUE,sep=";")
white <- read.csv("./data/wine_quality_white.csv",header = TRUE,sep=";")
red$type <- "red"
white$type <- "white"   
wine_quality <- rbind(red,white)
wine_quality$type <- factor(wine_quality$type)

接下来,我们根据 type 的值将数据分为测试和训练,这样每个数据帧都会得到一定比例的红酒和白葡萄酒,使用默认 caret::train() 设置训练数据和一个 GLM 方法。

library(caret)
set.seed(123)
inTrain <- createDataPartition(wine_quality$type, p = 3/4)[[1]]
training <- wine_quality[ inTrain,]
testing <- wine_quality[-inTrain,]

aModel <- train(type ~ .,data = training, method="glm", familia's = "binomial")

最后,我们使用模型对 hold back 数据帧进行预测,运行 一个混淆矩阵。

testLM <- predict(aModel,testing)
confusionMatrix(data=testLM,reference=testing$type)

...和输出:

> confusionMatrix(data=testLM,reference=testing$type)
Confusion Matrix and Statistics

          Reference
Prediction  red white
     red    393     3
     white    6  1221
                                          
               Accuracy : 0.9945          
                 95% CI : (0.9895, 0.9975)
    No Information Rate : 0.7542          
    P-Value [Acc > NIR] : <2e-16          
                                          
                  Kappa : 0.985           
                                          
 Mcnemar's Test P-Value : 0.505           
                                          
            Sensitivity : 0.9850          
            Specificity : 0.9975          
         Pos Pred Value : 0.9924          
         Neg Pred Value : 0.9951          
             Prevalence : 0.2458          
         Detection Rate : 0.2421          
   Detection Prevalence : 0.2440          
      Balanced Accuracy : 0.9913          
                                          
       'Positive' Class : red