2

我想使用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 或工作流中的哪个函数将渲染与(上面的第一个代码块)模型完全匹配的对象?

4

0 回答 0