混淆矩阵怎么画 使用R软件

混淆矩阵是一种统计工具，用于可视化分类模型的预测结果与实际标签之间的对应关系。在R语言中，我们可以使用`caret`包中的`confusionMatrix()`函数来创建混淆矩阵，并通过`ggplot2`包进行美化。

以下是绘制混淆矩阵的基本步骤：

1. 首先，确保你已经安装了`caret`和`ggplot2`库，如果没有，可以使用以下命令安装：

   install.packages("caret")
   install.packages("ggplot2")

2. 加载所需的包：

   library(caret)
   library(ggplot2)

3. 对于二分类问题，你可以使用`predict()`函数对测试数据进行预测，并将实际类别与预测类别作为输入到`confusionMatrix()`函数：

   # 假设你的模型名为model，测试集为test_data，实际标签为test_labels
   predictions <- predict(model, test_data)
   cm <- confusionMatrix(predictions, test_labels)

4. 现在你有了混淆矩阵对象`cm`，可以用它来绘制矩阵：

   cm_df <- as.data.frame(cm$table) # 转换为数据框以便处理
   ggplot(cm_df, aes(x=Reference, y=Predicted)) +
     geom_tile(fill = adjustcolor(cm$byClass$Overall["Accuracy"], alpha = 0.8), color = "white") +
     theme_minimal() +
     labs(title="Confusion Matrix", x="True Class", y="Predicted Class",
         fill="Proportion of Predictions", caption="@source{YourModelName}") +
     coord_fixed()

这段代码会生成一个比例条形图，颜色表示每个类别预测的准确性。

5. 最后别忘了保存或展示你的图形。