R/plot.rfCountData.R

In fpechon/rfCountData: Random Forests for Count Data

#' plot.rfCountData
#' @name plot.rfCountData
#' @param x, a rfCountData object
#' @param oob, Should the Poisson deviance of the OOB be plotted ? Defaults to \code{TRUE}.
#' @param testing, Should the Poisson deviance of the training set be plotted ? Defaults to \code{TRUE}, if a test set was used previously.
#' @export
plot.rfCountData <- function(x,  main=deparse(substitute(x)), oob=TRUE, testing=TRUE,...) {
  if(x$type == "unsupervised")
    stop("No plot for unsupervised randomForest.")
  test <- !is.null(x$test$dev) & testing 
  d = data.frame(ntree = 1:x$ntree, oob = x$dev, testing=NA)
  if (test)
    d$testing = x$test$dev
  
  p = ggplot2::ggplot(d)+ggplot2::xlab("trees") + ggplot2::ylab("Mean Poisson Deviance")
  if (oob)
  p = p + ggplot2::geom_line(ggplot2::aes(x = ntree, y = oob, colour = "Out-of-bag")) 
  if(test) {
    d$testing = x$test$dev
    p = p + ggplot2::geom_line(ggplot2::aes(x = ntree, y = testing, colour="Testing set"))
  }
    p +ggplot2::theme(legend.title = ggplot2::element_blank(), legend.position = "bottom")
}