R/plot.ps.cont.R

In twangContinuous: Toolkit for Weighting and Analysis of Nonequivalent Groups - Continuous Exposures

#' Plot the `ps.cont` object.
#'
#' This function produces a collection of diagnostic plots for `ps.cont` objects.
#'
#' @param x `ps.cont` object
#' @param plots An indicator of which type of plot is desired. The options are
#'   * `"optimize"` A plot of the balance criteria as a function of the GBM
#'     iteration.
#'   * `"es"` Plots of the standardized effect size of the pre-treatment
#'     variables before and after weighting
#' @param subset Used to restrict which of the `stop.method`s will be used
#'   in the figure.
#' @param ... Additional arguments.
#'
#' @return Returns diagnostic plots for `ps.cont` objects.
#'
#' @examples
#'    \dontrun{plot(test.mod)}
#'
#' @seealso [ps.cont]
#'
#' @method plot ps.cont
#' @export

plot.ps.cont <- function(x, plots="optimize", subset = NULL, ...) {
  pt1 <- diag.plot(x, plots, subset, ...)
  return(pt1)
}