R/explain_tidymodels.R
explain_tidymodels.RdDALEX is designed to work with various black-box models like tree ensembles, linear models, neural networks etc. Unfortunately R packages that create such models are very inconsistent. Different tools use different interfaces to train, validate and use models. One of those tools, which is one of the most popular one is the tidymodels package. We would like to present dedicated explain function for it.
object - a model to be explained
data.frame or matrix - data which will be used to calculate the explanations. If not provided, then it will be extracted from the model. Data should be passed without a target column (this shall be provided as the y argument). NOTE: If the target variable is present in the data, some of the functionalities may not work properly.
numeric vector with outputs/scores. If provided, then it shall have the same size as data
numeric vector with sampling weights. By default it's NULL. If provided, then it shall have the same length as data
function that takes two arguments: model and new data and returns a numeric vector with predictions. By default it is yhat.
Character or numeric containing either column name or column number in the model prediction object of the class that should be considered as positive (i.e. the class that is associated with probability 1). If NULL, the second column of the output will be taken for binary classification. For a multiclass classification setting, that parameter cause switch to binary classification mode with one vs others probabilities.
function that takes four arguments: model, data, target vector y and predict function (optionally). It should return a numeric vector with model residuals for given data. If not provided, response residuals (\(y-\hat{y}\)) are calculated. By default it is residual_function_default.
other parameters
character - the name of the model. By default it's extracted from the 'class' attribute of the model
logical. If TRUE (default) then diagnostic messages will be printed
logical. If TRUE (default) then predicted_values and residual are calculated when explainer is created.
This will happen also if verbose is TRUE. Set both verbose and precalculate to FALSE to omit calculations.
logical. If TRUE (default) then WARNINGS, ERRORS and NOTES are colorized. Will work only in the R console. Now by default it is FALSE while knitting and TRUE otherwise.
a named list (package, version, type) containing information about model. If NULL, DALEX will seek for information on it's own.
type of a model, either classification or regression. If not specified then type will be extracted from model_info.
explainer object (explain) ready to work with DALEX
library("DALEXtra")
library("tidymodels")
#> ── Attaching packages ────────────────────────────────────── tidymodels 0.2.0 ──
#> ✔ broom 0.8.0 ✔ rsample 0.1.1
#> ✔ dials 0.1.1 ✔ tibble 3.1.6
#> ✔ dplyr 1.0.8 ✔ tidyr 1.2.0
#> ✔ infer 1.0.0 ✔ tune 0.2.0
#> ✔ modeldata 0.1.1 ✔ workflows 0.2.6
#> ✔ parsnip 0.2.1 ✔ workflowsets 0.2.1
#> ✔ purrr 0.3.4 ✔ yardstick 0.0.9
#> ✔ recipes 0.2.0
#> ── Conflicts ───────────────────────────────────────── tidymodels_conflicts() ──
#> ✖ purrr::discard() masks scales::discard()
#> ✖ dplyr::explain() masks DALEX::explain()
#> ✖ dplyr::filter() masks stats::filter()
#> ✖ dplyr::lag() masks stats::lag()
#> ✖ recipes::step() masks stats::step()
#> • Search for functions across packages at https://www.tidymodels.org/find/
library("recipes")
data <- titanic_imputed
data$survived <- as.factor(data$survived)
rec <- recipe(survived ~ ., data = data) %>%
step_normalize(fare)
model <- decision_tree(tree_depth = 25) %>%
set_engine("rpart") %>%
set_mode("classification")
wflow <- workflow() %>%
add_recipe(rec) %>%
add_model(model)
model_fitted <- wflow %>%
fit(data = data)
explain_tidymodels(model_fitted, data = titanic_imputed, y = titanic_imputed$survived)
#> Preparation of a new explainer is initiated
#> -> model label : workflow ( default )
#> -> data : 2207 rows 8 cols
#> -> target variable : 2207 values
#> -> predict function : yhat.workflow will be used ( default )
#> -> predicted values : No value for predict function target column. ( default )
#> -> model_info : package tidymodels , ver. 0.2.0 , task classification ( default )
#> -> predicted values : numerical, min = 0.05555556 , mean = 0.3221568 , max = 0.9267399
#> -> residual function : difference between y and yhat ( default )
#> -> residuals : numerical, min = -0.9267399 , mean = 1.858834e-17 , max = 0.9444444
#> A new explainer has been created!
#> Model label: workflow
#> Model class: workflow
#> Data head :
#> gender age class embarked fare sibsp parch survived
#> 1 male 42 3rd Southampton 7.11 0 0 0
#> 2 male 13 3rd Southampton 20.05 0 2 0