【问题标题】:R: Tidymodels: Is it possible to plot the trees for a random forest model in tidy models?R:Tidymodels:是否可以在 tidy 模型中绘制随机森林模型的树?
【发布时间】:2021-08-11 23:14:51
【问题描述】:

是否可以在随机森林模型中绘制树木?以下是可用于解释的示例数据集。 对不起,我没有在网上找到任何这样的例子,因此我没有自己尝试任何东西。以下只是一个示例解决方法。

library(palmerpenguins)

penguins <- penguins %>%
  filter(!is.na(sex)) %>%
  select(-year, -island)

拆分数据

set.seed(123)
penguin_split <- initial_split(penguins, strata = sex)
penguin_train <- training(penguin_split)
penguin_test <- testing(penguin_split)

创建模型规范。

rf_spec <- rand_forest() %>%
  set_mode("classification") %>%
  set_engine("ranger")

penguin_wf <- workflow() %>%
  add_formula(sex ~ .)

应用于测试数据

penguin_final <- penguin_wf %>%
  add_model(rf_spec) %>%
  last_fit(penguin_split)

现在如何绘制树木?

提前致谢。

【问题讨论】:

    标签: r random-forest rpart tidymodels


    【解决方案1】:

    据我所知,没有绘制ranger 树或randomForest 树的内置函数(请参阅herehere)。然而,决策树的森林默认由 500 棵树组成,每棵树都有一个图似乎有些夸张。有一些方法可以从其他算法绘制决策树,例如rpartpartytree。看看here,简要了解这些绘制树木和森林的方法。

    【讨论】:

      猜你喜欢
      • 2018-09-26
      • 2020-02-25
      • 2019-11-22
      • 2019-07-10
      • 2017-08-11
      • 2015-09-23
      • 2016-04-09
      • 2012-06-10
      • 2020-12-15
      相关资源
      最近更新 更多