r - 如何使用从欧洲防风草函数 boost_tree 渲染的“_xgb.Booster”“model_fit”对象计算提升曲线和十分位数提升图表？

Question

我有这段代码可以呈现“_xgb.Booster”“model_fit”对象类。我应该，但我不确定如何在这里为整个可重现的示例代码提供数据！

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)

如何使用此 xgb 对象计算升力曲线和十分位升力图？

Answer 1

一旦您以示例数据集中的形式获得数据two_class_example，您就可以使用该lift_curve()函数来计算它。然后，您可以使用提升曲线进行可视化。

library(tidymodels)

two_class_lift <- two_class_example %>%
  lift_curve(truth, Class1) 

two_class_lift %>%
  autoplot()

two_class_lift %>%
  group_by(.percent_tested = cut_interval(.percent_tested, n = 10)) %>%
  summarise(.lift = mean(.lift, na.rm = TRUE)) %>%
  ggplot(aes(.percent_tested, .lift)) +
  geom_col() +
  theme_bw() +
  labs(x = "% Tested", y = "Lift")
#> `summarise()` ungrouping output (override with `.groups` argument)

^{由reprex 包于 2020-08-26 创建(v0.3.0.9001)}

如果您需要帮助从 xgboost 模型中获得预测，请查看这篇文章tidymodels.org。