Modélisation avec tidymodels en R
David Svancer
Data Scientist
Paramètres du modèle définis avant l'entraînement et qui contrôlent la complexité
parsnip decision tree
cost_complexitytree_depthmin_n
La fonction decision_tree() définit des valeurs d'hyperparamètres par défaut
cost_complexity est fixé à 0.01tree_depth est fixé à 30min_n est fixé à 20Ce ne sont pas forcément les meilleures valeurs pour tous les ensembles de données
dt_model <- decision_tree() %>%
set_engine('rpart') %>%
set_mode('classification')
La fonction tune() du paquet tune
tune() dans la spécification du modèle parsnipdt_tune_model <- decision_tree(cost_complexity = tune(), tree_depth = tune(), min_n = tune()) %>% set_engine('rpart') %>% set_mode('classification')dt_tune_model
Decision Tree Model Specification (classification)
Main Arguments:
cost_complexity = tune()
tree_depth = tune()
min_n = tune()
Computational engine: rpart
Les objets workflow se mettent à jour facilement
leads_wkfl précédentleads_wkfl à update_model() et fournissez un nouveau modèle d'arbre avec paramètres à ajusterleads_tune_wkfl <- leads_wkfl %>%update_model(dt_tune_model)leads_tune_wkfl
== Workflow ===============
Preprocessor: Recipe
Model: decision_tree()
-- Preprocessor -----------
3 Recipe Steps
* step_corr()
* step_normalize()
* step_dummy()
-- Model ------------------
Decision Tree Model Specification (classification)
Main Arguments: cost_complexity = tune()
tree_depth = tune()
min_n = tune()
Computational engine: rpart
Méthode la plus courante pour ajuster des hyperparamètres
| cost_complexity | tree_depth | min_n |
|---|---|---|
| 0.001 | 20 | 35 |
| 0.001 | 20 | 15 |
| 0.001 | 35 | 35 |
| 0.001 | 35 | 15 |
| 0.2 | 20 | 35 |
| ... | ... | ... |
La fonction parameters() du paquet dials
parsniptune(), le cas échéantdialsparameters(dt_tune_model)
Collection of 3 parameters for tuning
identifier type object
cost_complexity cost_complexity nparam[+]
tree_depth tree_depth nparam[+]
min_n min_n nparam[+]
Générer des combinaisons aléatoires
La fonction grid_random()
parameters()size fixe le nombre de combinaisons aléatoires à générerset.seed() avant grid_random() pour la reproductibilitéset.seed(214) grid_random(parameters(dt_tune_model),size = 5)
# A tibble: 5 x 3
cost_complexity tree_depth min_n
<dbl> <int> <int>
1 0.0000000758 14 39
2 0.0243 5 34
3 0.00000443 11 8
4 0.000000600 3 5
5 0.00380 5 36
Première étape de l'ajustement
dt_grid contient 5 combinaisons aléatoires d'hyperparamètresset.seed(214) dt_grid <- grid_random(parameters(dt_tune_model), size = 5)dt_grid
# A tibble: 5 x 3
cost_complexity tree_depth min_n
<dbl> <int> <int>
1 0.0000000758 14 39
2 0.0243 5 34
3 0.00000443 11 8
4 0.000000600 3 5
5 0.00380 5 36
La fonction tune_grid() effectue l'ajustement des hyperparamètres
Prend les arguments suivants :
workflow ou modèle parsnipresamplesgridmetrics optionnelleRetourne un tibble de résultats
.metricsdt_tuning <- leads_tune_wkfl %>%tune_grid(resamples = leads_folds,grid = dt_grid,metrics = leads_metrics)
dt_tuning
# Tuning results
# 10-fold cross-validation using stratification
# A tibble: 10 x 4
splits id .metrics ..
<list> <chr> <list> ..
<split [896/100]> Fold01 <tibble [15 x 7]> ..
................ ...... ............... ..
<split [897/99]> Fold09 <tibble [15 x 7]> ..
<split [897/99]> Fold10 <tibble [15 x 7]> ..
La fonction collect_metrics() fournit par défaut des résultats résumés
dt_tuning %>%
collect_metrics()
# A tibble: 15 x 9
cost_complexity tree_depth min_n .metric .estimator mean n std_err .config
<dbl> <int> <int> <chr> <chr> <dbl> <int> <dbl> <chr>
1 0.0000000758 14 39 roc_auc binary 0.827 10 0.0147 Model1
2 0.0000000758 14 39 sens binary 0.728 10 0.0277 Model1
3 0.0000000758 14 39 spec binary 0.865 10 0.0156 Model1
4 0.0243 5 34 roc_auc binary 0.823 10 0.0147 Model2
. ...... .. .. .... ...... ..... .. ..... ......
14 0.00380 5 36 sens binary 0.747 10 0.0209 Model5
15 0.00380 5 36 spec binary 0.858 10 0.0161 Model5
Modélisation avec tidymodels en R