iml (version 0.9.0)

predict.LocalModel: Predict LocalModel

Description

Predict the response for newdata with the LocalModel model.

Usage

# S3 method for LocalModel
predict(object, newdata = NULL, ...)

Arguments

object

A LocalModel R6 object

newdata

A data.frame for which to predict

...

Further arguments for the objects predict function

Value

A data.frame with the predicted outcome.

See Also

LocalModel

Examples

Run this code
# NOT RUN {
if (require("randomForest")) {
# First we fit a machine learning model on the Boston housing data
data("Boston", package  = "MASS")
X = Boston[-which(names(Boston) == "medv")]
rf = randomForest(medv ~ ., data = Boston, ntree = 50)
mod = Predictor$new(rf, data = X)

# Explain the first instance of the dataset with the LocalModel method:
x.interest = X[1,]
lemon = LocalModel$new(mod, x.interest = x.interest, k = 2)
predict(lemon, newdata = x.interest)
}
# }

Run the code above in your browser using DataLab