时间戳

首页 > 解决方案 > 如何使 tidymodels 或工作流结果与 DALEX 解释一起工作？

r - 如何使 tidymodels 或工作流结果与 DALEX 解释一起工作？

问题描述

我想使用modelStudio，为此我需要使用DALEX::explain()将模型对象作为参数来制作解释器。我应该，但我不确定如何在这里为整个可重现的示例代码提供数据！

当我使用以下代码时

    xgb <- boost_tree(mode = "classification",
                  trees = 100,
                  mtry = 0.7,
                  learn_rate = 0.15,
                  tree_depth = 10,
                  sample_size = 1) %>%
  set_engine("xgboost") %>%
  fit(Y ~ ., data = train)

以下解释器有效：

explainer <- explain(xgb,
                     data = test,
                     y = test$Y,
                     predict_function = , 
                     label = "xgb")

但是当我使用整个工作流程并尝试使用 提取模型pull_workflow_fit时，它会说它无法预测。

xgb1 <- final_xgb %>%
   fit(data = train) %>%
   pull_workflow_fit()

两种情况下的类都是_xgb.Booster model_fit.

xgbtidymodels 或工作流中的哪个函数将渲染与（上面的第一个代码块）模型完全匹配的对象？

标签： rtidymodelsdalex

解决方案

推荐阅读