# NOT RUN {
library("breakDown")
logit <- function(x) exp(x)/(1+exp(x))
HR_glm_model <- glm(left~., data = breakDown::HR_data, family = "binomial")
explainer_glm <- explain(HR_glm_model, data = HR_data, trans=logit)
expl_glm <- variable_response(explainer_glm, "satisfaction_level", "pdp")
expl_glm
# }
# NOT RUN {
library("randomForest")
HR_rf_model <- randomForest(factor(left)~., data = breakDown::HR_data, ntree = 100)
explainer_rf <- explain(HR_rf_model, data = HR_data,
predict_function = function(model, x)
predict(model, x, type = "prob")[,2])
expl_rf <- variable_response(explainer_rf, variable = "satisfaction_level", type = "pdp",
which.class = 2, prob = TRUE)
expl_rf
plot(expl_rf)
# }
# NOT RUN {
# }
Run the code above in your browser using DataCamp Workspace