CommencerCommencer gratuitement

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

Afficher le cours

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 = ____
Modifier et exécuter le code