R 的樹狀模型機器學習
Sandro Raabe
Data Scientist


library(tidymodels)
-- Attaching packages -------------------- tidymodels 0.1.4 --
v parsnip 0.2.1 v rsample 0.1.1
v dplyr 1.0.9 v tibble 3.1.7
v yardstick 0.0.9 v tune 0.1.6
1. 選擇模型類別
library(tidymodels)
decision_tree()
Decision Tree Model Specification (unknown)
2. 設定模型引擎
library(tidymodels)
decision_tree() %>%
set_engine("rpart")
Decision Tree Model Specification (unknown)
Computational engine: rpart
3. 設定模式(分類或迴歸)
library(tidymodels)
decision_tree() %>%
set_engine("rpart") %>%
set_mode("classification")
Decision Tree Model Specification (classification)
Computational engine: rpart
library(tidymodels)
tree_spec <- decision_tree() %>%
set_engine("rpart") %>%
set_mode("classification")
# 使用公式與訓練資料來擬合模型規格
tree_spec %>%
fit(formula = outcome ~ age + bmi,
data = diabetes)
parsnip model object
Fit time: 19 ms
n = 652
R 的樹狀模型機器學習