R语言二分类逻辑回归模型混淆矩阵可视化
时间: 2023-08-06 11:07:14 浏览: 156
混淆矩阵计算(适用二分类)
5星 · 资源好评率100%
您可以使用R语言中的`caret`包来计算二分类逻辑回归模型的混淆矩阵,并使用`ggplot2`包来进行可视化。以下是一个示例代码:
```R
# 导入所需的包
library(caret)
library(ggplot2)
# 创建一个示例数据集
data <- data.frame(X1 = rnorm(100), X2 = rnorm(100), Y = factor(sample(c(0, 1), 100, replace = TRUE)))
# 拆分数据集为训练集和测试集
set.seed(123)
trainIndex <- createDataPartition(data$Y, p = 0.7, list = FALSE)
trainData <- data[trainIndex, ]
testData <- data[-trainIndex, ]
# 训练逻辑回归模型
model <- train(Y ~ ., data = trainData, method = "glm", family = "binomial")
# 在测试集上进行预测
predictions <- predict(model, newdata = testData)
# 创建混淆矩阵
confusionMatrix <- caret::confusionMatrix(predictions, testData$Y)
# 可视化混淆矩阵
ggplot(confusionMatrix$table, aes(Prediction, Reference, fill = Freq)) +
geom_tile() +
geom_text(aes(label = Freq), color = "white") +
theme_minimal() +
scale_fill_gradient(low = "white", high = "steelblue") +
labs(title = "Confusion Matrix",
x = "Prediction",
y = "Reference")
```
这段代码首先导入了`caret`和`ggplot2`包,然后创建了一个示例数据集。接下来,使用`createDataPartition`函数将数据集拆分为训练集和测试集。然后,使用`train`函数训练逻辑回归模型,并使用`predict`函数在测试集上进行预测。随后,使用`confusionMatrix`函数计算混淆矩阵。最后,使用`ggplot`函数将混淆矩阵可视化为热力图。
请注意,这只是一个示例代码,您需要根据自己的数据和需求进行适当的修改。
阅读全文