Plot the predictor matrix of an imputation model

Usage

plot_pred(
  data,
  vrb = "all",
  method = NULL,
  label = TRUE,
  square = TRUE,
  rotate = FALSE
)

Arguments

data: A predictor matrix for mice, typically generated with mice::make.predictorMatrix or mice::quickpred.
vrb: String, vector, or unquoted expression with variable name(s), default is "all".
method: Character string or vector with imputation methods.
label: Logical indicating whether predictor matrix values should be displayed.
square: Logical indicating whether the plot tiles should be squares.
rotate: Logical indicating whether the variable name labels should be rotated 90 degrees.

Value

An object of class ggplot2::ggplot.

Examples

# generate a predictor matrix
pred <- mice::quickpred(mice::nhanes)

# plot predictor matrix for all columns
plot_pred(pred)


# plot predictor matrix for specific columns by supplying a character vector
plot_pred(pred, c("chl", "hyp"))


# plot predictor matrix for specific columns by supplying unquoted variable names
plot_pred(pred, c(chl, hyp))


# plot predictor matrix for specific columns by passing an object with variable names
# from the environment, unquoted with `!!`
my_variables <- c("chl", "hyp")
plot_pred(pred, !!my_variables)

# object with variable names must be unquoted with `!!`
try(plot_pred(pred, my_variables))
#> Error in match_vrb(vrb, row.names(data)) : 
#>   ✖ The variable name(s) supplied to `vrb` could not be found in `data`.
#> ℹ If you supply an object with variable names from the environment, use `!!` to
#>   unqote:
#>   `vrb = !!my_variables`