CommencerCommencer gratuitement

Modifier le nombre d’hyperparamètres à ajuster

Si l’on examine l’objet du modèle de près, on voit que caret a déjà effectué pour nous un réglage automatique des hyperparamètres : train crée automatiquement une grille de paramètres d’ajustement. Par défaut, si p est le nombre de paramètres d’ajustement, la taille de la grille est de 3^p. Mais nous pouvons aussi spécifier le nombre de valeurs différentes à essayer pour chaque hyperparamètre.

Les données ont de nouveau été préchargées sous le nom bc_train_data. Les bibliothèques caret et tictoc ont également été préchargées.

Cet exercice fait partie du cours

Optimisation des hyperparamètres en R

Afficher le cours

Instructions

  • Testez quatre valeurs différentes pour chaque hyperparamètre avec l’ajustement automatique dans caret.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Set seed.
set.seed(42)
# Start timer.
tic()
# Train model.
gbm_model <- train(diagnosis ~ ., 
                   data = bc_train_data, 
                   method = "gbm", 
                   trControl = trainControl(method = "repeatedcv", number = 5, repeats = 3),
                   verbose = FALSE,
                   ___)
# Stop timer.
toc()
Modifier et exécuter le code