inst/tinytest/enhances-randomForest.R

suppressPackageStartupMessages(library(randomForest))

# Continuous outcome
fit <- randomForest(Ozone ~ ., data=airquality, na.action=na.omit)
visreg(fit, "Temp")

# Binary outcome
mtcars$am <- factor(mtcars$am)
fit <- randomForest(am ~ ., mtcars)
visreg(fit, "wt")
pbreheny/visreg documentation built on March 20, 2024, 1:07 a.m.