R/lm.actualvfitted.ggplot.R

In Braja93/braja: General data manipulation functions

#' Plots actual v fitted
#'
#' Takes a lm.actualvfitted object and plots.
#' @param lm.contribution.object lm.contribution object. Output of lm.contribtion saved as an object.
#' @keywords lm actual fitted ggplot
#' @import data.table
#' @import ggplot2
#' @export

lm.actualvfitted.ggplot <- function(lm.actualvfitted.object) {
  
  id.var <- names(lm.actualvfitted.object)[1]
  dependent.var <- names(lm.actualvfitted.object)[2]
  options(warn = -1)
  out <- reshape2::melt(lm.actualvfitted.object, id.vars = id.var)
  options(warn = 0)
  ggplot2::ggplot(out, ggplot2::aes_string(x = id.var, y = "value", 
                                           colour = "variable", group = "variable")) +
    ggplot2::geom_line() +
    ggplot2::ylab(dependent.var)
  
}