Learn R Programming

DALEX (version 2.4.2)

predict.explainer: Predictions for the Explainer

Description

This is a generic predict() function works for explainer objects.

Usage

# S3 method for explainer
predict(object, newdata, ...)

model_prediction(explainer, new_data, ...)

Value

An numeric matrix of predictions

Arguments

object

a model to be explained, object of the class explainer

newdata

data.frame or matrix - observations for prediction

...

other parameters that will be passed to the predict function

explainer

a model to be explained, object of the class explainer

new_data

data.frame or matrix - observations for prediction

Examples

Run this code
HR_glm_model <- glm(status == "fired"~., data = HR, family = "binomial")
explainer_glm <- explain(HR_glm_model, data = HR)
predict(explainer_glm, HR[1:3,])

 # \donttest{
library("ranger")
HR_ranger_model <- ranger(status~., data = HR, num.trees = 50, probability = TRUE)
explainer_ranger  <- explain(HR_ranger_model, data = HR)
predict(explainer_ranger, HR[1:3,])

model_prediction(explainer_ranger, HR[1:3,])
 # }

Run the code above in your browser using DataLab