R语言二分类混淆矩阵可视化

你可以使用R语言中的`caret`和`ggplot2`库来可视化二分类混淆矩阵。以下是一个简单的示例代码：

# 导入必要的库
library(caret)
library(ggplot2)

# 创建一个简单的二分类混淆矩阵
actual <- c(1, 0, 0, 1, 1, 1)
predicted <- c(1, 0, 1, 1, 0, 1)
confusion_matrix <- confusionMatrix(predicted, actual)

# 提取混淆矩阵的数据
cm_data <- as.data.frame(confusion_matrix$table)
names(cm_data) <- c("Predicted", "Actual", "Count")

# 绘制混淆矩阵的热力图
ggplot(cm_data, aes(x = Predicted, y = Actual, fill = Count)) +
  geom_tile() +
  geom_text(aes(label = Count), color = "white", size = 12) +
  scale_fill_gradient(low = "white", high = "steelblue") +
  theme_minimal()

这段代码首先使用`caret`库中的`confusionMatrix`函数创建一个二分类混淆矩阵对象。然后，使用`ggplot2`库来绘制热力图，其中`geom_tile`用于绘制矩形，`geom_text`用于在矩形中显示计数，`scale_fill_gradient`用于设置填充颜色。最后，使用`theme_minimal`来设置图形主题。

你可以根据自己的实际数据和需求进行适当的修改和扩展。希望对你有所帮助！