R/diagnostic_plots.R

In jennguyen1/jn.general: Generic functions I use

#' Diagnostic Plots for lm and glm
#'
#' Generates the diagnostic plots in ggplot2, similar to the ones that are generated via the plot command. Currently available for lm and glm models
#'
#' @param mod a lm or glm model obj
#'
#' @return a list of diagnostic plots
#' 
#' @import ggplot2
#' @export

diagnostic_plots <- function(mod){
  "Generates diagnostic plots for lm and glm models"
  
  assertthat::assert_that(!missing(mod), msg = "Input mod is missing")
  assertthat::assert_that( any(class(mod) %in% c("lm", "glm")), msg = "Invalid mod argument")
  if( any(class(mod) %in% "glm") ) {
    return( glm_plot(mod) )
  } else if( class(mod) %in% "lm" ) {
    return( lm_plot(mod) )
  }
}