6

パッケージを使用して意思決定ツリー モデルを構築できましたtidymodelsが、結果を取得してツリーをプロットする方法がわかりません。rpartとパッケージを使用して同じことを達成できることは知っていますが、それが私が学んでいることなので、rpart.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をプロットできる形式に変換するために欠落しているパッケージまたはステップがあるかどうかはわかりません。これは不可能かもしれませんが、どんな助けでも大歓迎です。

4

2 に答える 2