Explorez un espace de modèles plus large

Machine Learning avec caret en R

Zach Mayer

Data Scientist at DataRobot and co-author of caret

Les forêts aléatoires nécessitent un réglage

Les hyperparamètres contrôlent l'ajustement du modèle
Choisis « à la main » avant l'ajustement
Le plus important est mtry
- Nombre de variables sélectionnées aléatoirement à chaque division
Valeur faible = plus aléatoire
Valeur élevée = moins aléatoire
Difficile de connaître la meilleure valeur à l'avance

Exemple : données sonar

Argument tuneLength de caret::train()
Indique à caret combien de variantes essayer

# Load some data
library(caret)
library(mlbench)
data(Sonar)

# Fit a model with a deeper tuning grid
model <- train(
  Class ~ ., 
  data = Sonar, 
  method = "ranger", 
  tuneLength = 10
)

# Plot the results
plot(model)

Tracer les résultats

L'image montre une courbe de l'exactitude bootstrap en fonction du nombre de paramètres sélectionnés aléatoirement. Elle atteint un pic vers x = 14, puis chute rapidement.

Passons à la pratique !

Machine Learning avec caret en R