View source: R/TreeSurrogate.R
predict.TreeSurrogate | R Documentation |
Predict the response for newdata of a TreeSurrogate object.
This function makes the TreeSurrogate object call
its internal $predict()
method.
## S3 method for class 'TreeSurrogate'
predict(object, newdata, type = "prob", ...)
object |
The surrogate tree. A TreeSurrogate object. |
newdata |
A data.frame for which to predict. |
type |
Either "prob" or "class". Ignored if the surrogate tree does regression. |
... |
Further arguments for |
A data.frame with the predicted outcome.
In case of regression it is the predicted \hat{y}
. In case of
classification it is either the class probabilities (for type "prob") or the
class label (type "class")
TreeSurrogate
library("randomForest")
# Fit a Random Forest on the Boston housing data set
data("Boston", package = "MASS")
rf <- randomForest(medv ~ ., data = Boston, ntree = 50)
# Create a model object
mod <- Predictor$new(rf, data = Boston[-which(names(Boston) == "medv")])
# Fit a decision tree as a surrogate for the whole random forest
dt <- TreeSurrogate$new(mod)
# Plot the resulting leaf nodes
predict(dt, newdata = Boston)
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.