tests/test.earth.R

In earth: Multivariate Adaptive Regression Splines

# test.earth.R
# Check for porting problems by building a few simple models.
# For much more comprehensive tests see earth\inst\slowtests.
library(earth)
options(digits=3) # prevent floating point implementation issues across machines
data(trees)
earth.mod <- earth(Volume~., data=trees)
print(summary(earth.mod))
allowed.func <- function(degree, pred, parents, namesx)
{
    namesx[pred] != "Height"  # disallow "Height"
}
set.seed(2019)
earth.mod2 <- earth(Volume~., data=trees, allowed=allowed.func, trace=1)
print(summary(earth.mod2))
# multiple response model using class "Formula" and a factor predictor
data(iris)
earth.mod3 <- earth(Sepal.Length + Sepal.Width ~ Species, data=iris)
print(summary(earth.mod3))
plot(earth.mod3, nresponse="Sepal.Length", which=c(1,3), do.par=2, legend.pos="topleft")
plotmo(earth.mod3, nresponse=1, pt.col=2, do.par=0)