| name | iml |
| description | R iml package for interpretable ML. Use for model-agnostic interpretability methods. |
iml
Interpretable Machine Learning.
Create Predictor
library(iml)
predictor <- Predictor$new(
model = model,
data = train_data,
y = train_labels
)
Feature Importance
imp <- FeatureImp$new(predictor, loss = "mse")
plot(imp)
imp$results
Partial Dependence
pdp <- FeatureEffect$new(predictor, feature = "age")
plot(pdp)
pdp <- FeatureEffect$new(predictor, feature = "age",
method = "pdp")
pdp <- FeatureEffect$new(predictor, feature = "age",
method = "ale")
pdp <- FeatureEffect$new(predictor, feature = "age",
method = "pdp+ice")
Feature Interactions
interact <- Interaction$new(predictor)
plot(interact)
interact <- Interaction$new(predictor, feature = "age")
SHAP Values
shap <- Shapley$new(predictor, x.interest = new_data[1, ])
plot(shap)
shap$results
LIME
lime <- LocalModel$new(predictor, x.interest = new_data[1, ])
plot(lime)
lime$results
Surrogate Model
tree <- TreeSurrogate$new(predictor, maxdepth = 3)
plot(tree)
tree$predict(new_data)
Counterfactuals
library(counterfactuals)
cf <- Counterfactuals$new(predictor, x.interest = new_data[1, ])
cf$find_counterfactuals(desired_outcome = 1)
Multiple Features
pdp2d <- FeatureEffect$new(predictor,
feature = c("age", "income"),
method = "pdp")
plot(pdp2d)