#' 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")
}
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.