[英]How to load already computed confusion matrix to be used with the caret package in R
[英]How to plot confusion matrix in R with caret package
我如何 plot R 中的顏色混淆矩陣? 我有混淆矩陣統計信息,但我不知道如何將 plot 轉換為可視化 plot。 任何人請協助。 謝謝。
# Training Naive Bayes model using {caret} package with 10 fold cross validation
NBclassifierCaretCV <- train(x=data[, -1], y=data$diagnosis, 'nb',
trControl=trainControl(method='cv', number=10))
CVtrainDataset <- predict(NBclassifierCaretCV, newdata=data[, -1])
# Confusion matrix and a summary / using caret package
confusionMatrix(data=CVtrainDataset, data$diagnosis)
假設您的混淆矩陣如下:
pred <- factor(sample(1:5, 100, replace = T))
ref<- factor(sample(1:5, 100, replace = T))
library(caret)
cm <- confusionMatrix(pred, ref)
然后,您會想要檢索/創建一些信息,例如:
library(ggplot2)
library(grid)
library(gridExtra)
library(likert)
cm <- confusionMatrix(pred,ref) #create a confusion matrix
cm_d <- as.data.frame(cm$table) # extract the confusion matrix values as data.frame
cm_st <-data.frame(cm$overall) # confusion matrix statistics as data.frame
cm_st$cm.overall <- round(cm_st$cm.overall,2) # round the values
cm_d$diag <- cm_d$Prediction == cm_d$Reference # Get the Diagonal
cm_d$ndiag <- cm_d$Prediction != cm_d$Reference # Off Diagonal
cm_d[cm_d == 0] <- NA # Replace 0 with NA for white tiles
cm_d$Reference <- reverse.levels(cm_d$Reference) # diagonal starts at top left
cm_d$ref_freq <- cm_d$Freq * ifelse(is.na(cm_d$diag),-1,1)
繪制矩陣
plt1 <- ggplot(data = cm_d, aes(x = Prediction , y = Reference, fill = Freq))+
scale_x_discrete(position = "top") +
geom_tile( data = cm_d,aes(fill = ref_freq)) +
scale_fill_gradient2(guide = FALSE ,low="red3",high="orchid4", midpoint = 0,na.value = 'white') +
geom_text(aes(label = Freq), color = 'black', size = 3)+
theme_bw() +
theme(panel.grid.major = element_blank(), panel.grid.minor = element_blank(),
legend.position = "none",
panel.border = element_blank(),
plot.background = element_blank(),
axis.line = element_blank(),
)
繪制統計數據
plt2 <- tableGrob(cm_st)
全部一起
grid.arrange(plt1, plt2, nrow = 1, ncol = 2,
top=textGrob("Confusion Matrix",gp=gpar(fontsize=25,font=1)))
這不是最漂亮的 plot 但您可以稍后更改圖形組件。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.