【发布时间】: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