标签: dalex

这个问题指的是在 R 中使用 tidymodels 获取 catboost 模型的摘要形状图。鉴于问题下方的评论，OP 找到了解决方案，但到目前为止尚未与社区分享。

我想分析我的带有tidymodelsSHAP 值图的软件包的树整体，例如单个观测值的图，例如

并总结我的数据集所有特征的影响，例如

DALEXtra 提供为 tidymodels 创建 SHAP 值的函数explain.tidymodels()。 force_plot该fastshap包为底层 python 包的绘图函数提供了一个包装器SHAP。但我无法理解如何使该函数与函数的输出一起工作explain.tidymodels()。

问题：如何在 R 中使用tidymodels和生成这样的 SHAP 图explain.tidymodels？

MWE（对于带有 的 SHAP 值explain.tidymodels）

library(MASS)
library(tidyverse)
library(tidymodels)
library(parsnip)
library(treesnip)
library(catboost)
library(fastshap)
library(DALEXtra)
set.seed(1337)
rec <-  recipe(crim ~ ., data = Boston)

split <- initial_split(Boston)

train_data <- training(split)

test_data <- testing(split) %>% dplyr::select(-crim) %>% as.matrix()

model_default<-
  parsnip::boost_tree( …