Partial dependence plots绘制问题:标题遮挡、颜色参数修改失效
1. 解决文字遮挡问题
绘图底部的“Created for the workflow model”是默认自动添加的图注(caption),遮挡表头是默认布局边距不合理导致的,可选择以下任意一种方案:
- 直接移除不需要的图注:
model_profile(explainer, variables = c("AC", "Jaar", "Month", "Retentie")) %>% plot() + labs(caption = NULL)
- 保留图注但调整位置避免遮挡:
model_profile(explainer, variables = c("AC", "Jaar", "Month", "Retentie")) %>% plot() + theme(plot.caption = element_text(margin = margin(t = 15)))
t=15是调整图注的顶部边距,数值可以根据实际显示效果调整大小。
2. 调整线条颜色为红色
直接向plot()传入color/col参数不生效,是因为DALEX的plot.model_profile方法不支持直接通过该参数修改单条PDP线的颜色,直接叠加ggplot2的颜色 scale 即可生效:
model_profile(explainer, variables = c("AC", "Jaar", "Month", "Retentie")) %>% plot() + scale_color_manual(values = "red")
如果后续需要绘制多分组的偏依赖图,也可以通过values参数传入命名向量,为不同分组指定对应的颜色。
内容的提问来源于stack exchange,提问作者Kylian
相关产品推荐
相关产品推荐

