1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
% Generated by roxygen2: do not edit by hand
% Please edit documentation in R/bagEarth.R, R/bagFDA.R
\name{predict.bagEarth}
\alias{predict.bagEarth}
\alias{predict.bagFDA}
\title{Predicted values based on bagged Earth and FDA models}
\usage{
\method{predict}{bagEarth}(object, newdata = NULL, type = NULL, ...)
\method{predict}{bagFDA}(object, newdata = NULL, type = "class", ...)
}
\arguments{
\item{object}{Object of class inheriting from \code{bagEarth}}
\item{newdata}{An optional data frame or matrix in which to look for
variables with which to predict. If omitted, the fitted values are used
(see note below).}
\item{type}{The type of prediction. For bagged \code{\link[earth]{earth}}
regression model, \code{type = "response"} will produce a numeric vector of
the usual model predictions. \code{\link[earth]{earth}} also allows the user
to fit generalized linear models. In this case, \code{type = "response"}
produces the inverse link results as a vector. In the case of a binomial
generalized linear model, \code{type = "response"} produces a vector of
probabilities, \code{type = "class"} generates a factor vector and
\code{type = "prob"} produces a two-column matrix with probabilities for
both classes (averaged across the individual models). Similarly, for bagged
\code{\link[mda]{fda}} models, \code{type = "class"} generates a factor
vector and \code{type = "probs"} outputs a matrix of class probabilities.}
\item{\dots}{not used}
}
\value{
A vector of predictions (for regression or \code{type = "class"})
or a data frame of class probabilities. By default, when the model
predicts a number, a vector of numeric predictions is returned. When
a classification model is used, the default prediction is a factor vector
of classes.
}
\description{
Predicted values based on bagged Earth and FDA models
}
\note{
If the predictions for the original training set are needed, there are
two ways to calculate them. First, the original data set can be predicted by
each bagged earth model. Secondly, the predictions from each bootstrap
sample could be used (but are more likely to overfit). If the original call
to \code{bagEarth} or \code{bagFDA} had \code{keepX = TRUE}, the first
method is used, otherwise the values are calculated via the second method.
}
\examples{
\dontrun{
data(trees)
## out of bag predictions vs just re-predicting the training set
set.seed(655)
fit1 <- bagEarth(Volume ~ ., data = trees, keepX = TRUE)
set.seed(655)
fit2 <- bagEarth(Volume ~ ., data = trees, keepX = FALSE)
hist(predict(fit1) - predict(fit2))
}
}
\seealso{
\code{\link{bagEarth}}
}
\author{
Max Kuhn
}
\keyword{regression}
|
