R/plot_rfa.R

In milesdwilliams15/RFA: Estimate Average Treatment Effect via Random Forest Residualization

#' Wrapper for ggplot2 to Plot Histogram of Empirical ATE Replicates
#'
#' This function takes output from a `rfa` object and returns a
#' histogram of the bootstrapped ATE replicates.
#'
#' @export
plot_rfa = function(model, varname = "predictor"){
  tidy(model$fit) %>%
    filter(term == "xres") %>%
    ggplot() +
    aes(
      x = estimate,
      y = varname,
      xmin = conf.low,
      xmax = conf.high
    ) +
    geom_point() +
    geom_errorbarh(height = 0)
}