CommencerCommencer gratuitement

Ajuster une forêt aléatoire avec un réglage personnalisé

Maintenant que vous avez exploré les grilles de réglage par défaut fournies par la fonction train(), allons plus loin en personnalisant vos modèles.

Vous pouvez fournir autant de valeurs que vous le souhaitez pour mtry, de 2 jusqu’au nombre de colonnes du jeu de données. En pratique, les gains deviennent marginaux pour des valeurs très élevées de mtry. Vous allez donc utiliser une grille personnalisée qui explore 2 modèles simples (mtry = 2 et mtry = 3) ainsi qu’un modèle plus complexe (mtry = 7).

Cet exercice fait partie du cours

Machine Learning avec caret en R

Afficher le cours

Exercice interactif pratique

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

# Define the tuning grid: tuneGrid
tuneGrid <- data.frame(
  .mtry = ___,
  .splitrule = "___",
  .min.node.size = ___
)
Modifier et exécuter le code