gpt4 book ai didi

r - 从 tidymodels 绘制决策树结果

转载 作者:行者123 更新时间:2023-12-03 20:24:43 28 4
gpt4 key购买 nike

我已经设法使用 tidymodels 构建了一个决策树模型。包,但我不确定如何提取结果并绘制树。我知道我可以使用 rpartrpart.plot包来实现同样的事情,但我宁愿使用 tidymodels因为这就是我正在学习的。以下是使用 mtcars 的示例数据。

library(tidymodels)
library(rpart)
library(rpart.plot)
library(dplyr) #contains mtcars

#data
df <- mtcars %>%
mutate(gear = factor(gear))


#train/test
set.seed(1234)

df_split <- initial_split(df)
df_train <- training(df_split)
df_test <- testing(df_split)


df_recipe <- recipe(gear~ ., data = df) %>%
step_normalize(all_numeric())


#building model
tree <- decision_tree() %>%
set_engine("rpart") %>%
set_mode("classification")

#workflow
tree_wf <- workflow() %>%
add_recipe(df_recipe) %>%
add_model(tree) %>%
fit(df_train) #results are found here

rpart.plot(tree_wf$fit$fit) #error is here
我得到的错误是 Error in rpart.plot(tree_wf$fit$fit) : Not an rpart object这是有道理的,但我不知道我是否缺少将结果转换为 rpart.plot 格式的包或步骤。将允许我绘图。这可能是不可能的,但任何帮助将不胜感激。

最佳答案

您也可以使用 workflows::pull_workflow_fit()功能。它使代码更优雅一点。

tree_fit <- tree_wf %>% 
pull_workflow_fit()
rpart.plot(tree_fit$fit)

关于r - 从 tidymodels 绘制决策树结果,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63527684/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com