fviz_silhouette
我建议使用附加var.col
输入参数的以下修改版本。
myfviz_silhouette <- function (sil.obj, var.col, label = FALSE, print.summary = TRUE, ...) {
if (inherits(sil.obj, c("eclust", "hcut", "pam", "clara",
"fanny"))) {
df <- as.data.frame(sil.obj$silinfo$widths, stringsAsFactors = TRUE)
}
else if (inherits(sil.obj, "silhouette"))
df <- as.data.frame(sil.obj[, 1:3], stringsAsFactors = TRUE)
else stop("Don't support an oject of class ", class(sil.obj))
df <- df[order(df$cluster, -df$sil_width), ]
if (!is.null(rownames(df)))
df$name <- factor(rownames(df), levels = rownames(df))
else df$name <- as.factor(1:nrow(df))
df$cluster <- as.factor(df$cluster)
df$var_col <- var.col
mapping <- aes_string(x = "name", y = "sil_width", color = "var_col",
fill = "var_col")
p <- ggplot(df, mapping) + geom_bar(stat = "identity") +
labs(y = "Silhouette width Si", x = "", title = paste0("Clusters silhouette plot ",
"\n Average silhouette width: ", round(mean(df$sil_width),
2))) + ggplot2::ylim(c(NA, 1)) + geom_hline(yintercept = mean(df$sil_width),
linetype = "dashed", color = "red")
p <- ggpubr::ggpar(p, ...)
if (!label)
p <- p + theme(axis.text.x = element_blank(), axis.ticks.x = element_blank())
else if (label)
p <- p + theme(axis.text.x = element_text(angle = 45))
ave <- tapply(df$sil_width, df$cluster, mean)
n <- tapply(df$cluster, df$cluster, length)
sil.sum <- data.frame(cluster = names(ave), size = n, ave.sil.width = round(ave,
2), stringsAsFactors = TRUE)
if (print.summary)
print(sil.sum)
p
}
以下是它的使用示例:
library(factoextra)
library(cluster)
pamspec <- pam(iris[,-5], 3, keep.diss = TRUE)
color_var <- iris$Species
myfviz_silhouette(pamspec, color_var, label=TRUE) +
theme_classic()
