Rで学ぶTree-Based ModelsによるMachine Learning
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. モードを設定(分類 or 回帰)
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で学ぶTree-Based ModelsによるMachine Learning