Learn R Programming

cjbart (version 0.3.2)

plot.cjbart.vimp: Plot Variable Importance Matrix for Heterogeneity Analysis

Description

Plots a heatmap of variable importance, across predicted IMCEs. By default, all attribute-levels and covariates in the model are plotted.

Usage

# S3 method for cjbart.vimp
plot(x, covars = NULL, att_levels = NULL, ...)

Value

Plot of covariate importance scores

Arguments

x

Object of class cjbart.vimp, the result of running het_vimp()

covars

Optional vector of covariate names to plot. By default, all included covariates are shown.

att_levels

Optional vector of attribute-levels to plot. By default, all attribute-levels are shown.

...

Additional arguments (not currently used)