plot_correlation {DataExplorer} | R Documentation |
Create correlation heatmap for discrete features
Description
This function creates a correlation heatmap for all discrete categories.
Usage
plot_correlation(
data,
type = c("all", "discrete", "continuous"),
maxcat = 20L,
cor_args = list(),
geom_text_args = list(),
title = NULL,
ggtheme = theme_gray(),
theme_config = list(legend.position = "bottom", axis.text.x = element_text(angle = 90))
)
Arguments
data |
input data |
type |
column type to be included in correlation calculation. "all" for all columns, "discrete" for discrete features, "continuous" for continuous features. |
maxcat |
maximum categories allowed for each discrete feature. The default is 20. |
cor_args |
a list of other arguments to cor |
geom_text_args |
a list of other arguments to geom_text |
title |
plot title |
ggtheme |
complete ggplot2 themes. The default is theme_gray. |
theme_config |
a list of configurations to be passed to theme. |
Details
For discrete features, the function first dummifies all categories, then calculates the correlation matrix (see cor) and plots it.
Value
invisibly return the ggplot object
Examples
plot_correlation(iris)
plot_correlation(iris, type = "c")
plot_correlation(airquality, cor_args = list("use" = "pairwise.complete.obs"))
[Package DataExplorer version 0.8.3 Index]