predict.bagEarth {caret} | R Documentation |
Predicted values based on bagged Earth and FDA models
predict.bagEarth(object, newdata = NULL, type = "response", ...) predict.bagFDA(object, newdata = NULL, type = "class", ...)
object |
Object of class inheriting from bagEarth |
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). |
type |
The type of prediction. For bagged earth regression model, type = "response" will produce a numeric vector of the usual model predictions. earth also allows the user to fit generalized linear models. In this case, type = "response" produces the inverse link results as a vector. In the case of a binomial generalized linear model, type = "response" produces a vector of probabilities, type = "class" generates a factor vector and type = "prob" produces a 2 column matrix with probabilities for both classes (averaged across the individual models). Similarly, for bagged fda models, type = "class" generates a factor vector and type = "probs" outputs a matrix of class probabilities. |
... |
not used |
a vector of predictions
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 form each bootstrap sample could be used (but are more likely to overfit). If the original call to bagEarth
or bagFDA
had keepX = TRUE
, the first method is used, otherwise the values are calculated via the second method.
Max Kuhn
data(trees) fit1 <- bagEarth(Volume ~ ., data = trees, keepX = TRUE) fit2 <- bagEarth(Volume ~ ., data = trees, keepX = FALSE) hist(predict(fit1) - predict(fit2))