CommencerCommencer gratuitement

Ajuster la pénalité

Convaincu que le Lasso est une approche judicieuse pour réduire le nombre de variables de votre modèle tout en maintenant des performances acceptables, vous souhaitez ajuster le modèle en choisissant la meilleure valeur de pénalité. Une recipe de base ainsi que les partitions train et test sont chargées dans votre environnement.

Cet exercice fait partie du cours

Feature engineering en R

Afficher le cours

Instructions

  • Configurez votre modèle afin que la pénalité soit ajustée automatiquement.
  • Définissez une grille de pénalité avec 30 niveaux.

Exercice interactif pratique

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

# Set up your model so that the penalty is tuned automatically
model_lasso_tuned <- logistic_reg() %>% set_engine("glmnet") %>%
  set_args(mixture = 1, ___ = ___) 
workflow_lasso_tuned <- workflow() %>%
  add_model(model_lasso_tuned) %>%
  add_recipe(recipe)

# Configure a penalty grid with 30 levels
penalty_grid <- grid_regular(penalty(range = c(-3, 1)), ___ = ___)

tune_output <- tune_grid(workflow_lasso_tuned,
  resamples = vfold_cv(train, v = 5),
  metrics = metric_set(roc_auc),grid = penalty_grid)

autoplot(tune_output)
Modifier et exécuter le code