Expérimenter avec le dropout
L’exclusion permet d’éviter le surajustement en définissant aléatoirement certaines valeurs de sortie sur zéro pendant l’apprentissage. Dans cet exercice, vous allez construire un réseau neuronal simple avec dropout et observer comment il se comporte dans les modes d’apprentissage et d’évaluation.
Le paquet torch.nn
est préchargé en tant que nn
, et features
est déjà défini pour vous.
Cet exercice fait partie du cours
Introduction au deep learning avec PyTorch
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Model with Dropout
model = nn.Sequential(
nn.Linear(8, 6),
nn.Linear(6, 4),
____)
# Forward pass in training mode (Dropout active)
model.____
output_train = ____