Error in FI/PDP on model with one variable

[hbaniecki]

related to: https://github.com/ModelOriented/modelStudio/issues/99

[hbaniecki]

library(DALEX)
titanic_imputed_sub <- subset(titanic_imputed, select = c("survived", "class"))
model_titanic <- glm(survived ~ class, data = titanic_imputed_sub, family = "binomial")

# create an explainer for the model
explainer_titanic <- explain(model_titanic,
                             data = titanic_imputed_sub[, 2, drop=FALSE],
                             y = titanic_imputed$survived,
                             label = "Titanic GLM")

fi <- ingredients::feature_importance(explainer_titanic$model,
                                      explainer_titanic$data,
                                      explainer_titanic$y)

plot(fi)

pdp <- ingredients::partial_dependence(explainer_titanic$model,
                                       explainer_titanic$data)

plot(pdp)

[pbiecek]

thanks