Machine Learning avec des modèles arborescents en R
Sandro Raabe
Data Scientist
ranger, randomForesttidymodels pour ces forfaits : rand_forest() (dans le forfait parsnip)
rand_forest()Hyperparamètres :
mtry : prédicteurs vus à chaque nœud, valeur par défaut :trees : nombre d'arbres dans la forêt min_n : taille minimale permise d'un nœudrand_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")
Spécification du modèle Random Forest(classification)Arguments principaux : trees = 100Moteur de calcul : ranger
spec %>% fit(still_customer ~ ., data = customers_train)
objet modèle parsnip
Temps d'ajustement : 631 ms
Résultat Ranger
Nombre d'arbres : 100
Taille de l'échantillon : 9116
Nombre de variables indépendantes : 19
Mtry : 4
Taille cible des nœuds : 10
rand_forest(mode = "classification") %>% set_engine("ranger", importance = "impurity") %>%fit(still_customer ~ ., data = customers_train) %>%vip::vip()

Machine Learning avec des modèles arborescents en R