# NOT RUN {
HR$evaluation <- factor(HR$evaluation)
HR_glm_model <- glm(status == "fired"~., data = HR, family = "binomial")
explainer_glm <- explain(HR_glm_model, data = HR)
expl_glm <- variable_response(explainer_glm, "age", "pdp")
plot(expl_glm)
# }
# NOT RUN {
library("randomForest")
HR_rf_model <- randomForest(status == "fired" ~., data = HR)
explainer_rf <- explain(HR_rf_model, data = HR)
expl_rf <- variable_response(explainer_rf, variable = "age",
type = "pdp")
plot(expl_rf)
plot(expl_rf, expl_glm)
# Example for factor variable (with factorMerger)
expl_rf <- variable_response(explainer_rf, variable = "evaluation", type = "factor")
plot(expl_rf)
expl_glm <- variable_response(explainer_glm, variable = "evaluation", type = "factor")
plot(expl_glm)
# both models
plot(expl_rf, expl_glm)
# }
# NOT RUN {
# }
Run the code above in your browser using DataLab