Learn R Programming

iml (version 0.1)

predict.TreeSurrogate: Surrogate tree prediction

Description

Predict the response for newdata with the surrogate tree

Usage

# S3 method for TreeSurrogate
predict(object, newdata, type = "prob", ...)

Arguments

object

The surrogate tree. A TreeSurrogate R6 object

newdata

A data.frame for which to predict

type

Either "prob" or "class". Ignored if the surrogate tree does regression.

...

Further argumets for predict_party

Value

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")

Details

This function makes the TreeSurrogate object call its iternal object$predict() method. For examples see tree.surrogate

See Also

tree.surrogate