使用R语言绘制混淆矩阵热力图

使用R语言绘制混淆矩阵热力图

混淆矩阵是机器学习中常用的评估分类模型性能的工具。它展示了模型对各个类别样本的分类情况，有助于我们了解模型的准确性、召回率和精确度等指标。为了更直观地展示混淆矩阵的结果，我们可以使用R语言中的autoplot函数绘制一个热力图。

首先，我们需要安装并加载ggplot2和caret两个包，它们提供了绘图和机器学习相关的函数和工具。

# 安装包
install.packages("ggplot2")
install.packages("caret")

# 加载包
library(ggplot2)
library(caret)

接下来，假设我们有一个已经构建好的混淆矩阵confusion_matrix，它是一个二维矩阵，行和列分别表示真实类别和预测类别。我们可以使用autoplot函数将其可视化为热力图。

# 创建混淆矩阵
confusion_matrix <- matrix(c(50, 3, 7, 40), nrow = 2)
dimnames(confusion_matrix) <- list(Actual = c("Positive", "Negative"),