Uczenie maszynowe z modelami drzewiastymi w R
Sandro Raabe
Data Scientist
ranger, randomForesttidymodels: rand_forest() (pakiet parsnip)
rand_forest()Hiperparametry:
mtry: predyktory rozważane w każdym węźle, domyślnie:trees: liczba drzew w lesie min_n: minimalna liczba obserwacji w węźlerand_forest(mtry = 4,trees = 500,min_n = 10) %>%# Set the mode set_mode("classification") %>%# Use engine ranger or randomForest set_engine("ranger")
spec <- rand_forest(trees = 100) %>%set_mode("classification") %>%set_engine("ranger")
Random Forest Model Specification(classification)Main Arguments: trees = 100Computational engine: ranger
spec %>% fit(still_customer ~ ., data = customers_train)
parsnip model object
Fit time: 631ms
Ranger result
Number of trees: 100
Sample size: 9116
Number of independent variables: 19
Mtry: 4
Target node size: 10
rand_forest(mode = "classification") %>% set_engine("ranger", importance = "impurity") %>%fit(still_customer ~ ., data = customers_train) %>%vip::vip()

Uczenie maszynowe z modelami drzewiastymi w R