CommencezCommencez gratuitement

Expérimenter avec le dropout

Le dropout aide à prévenir le surapprentissage en mettant aléatoirement certaines valeurs de sortie à zéro pendant l'entraînement. Dans cet exercice, vous allez créer un réseau de neurones simple avec dropout et observer son comportement en modes entraînement et évaluation.

Le module torch.nn est préchargé sous nn, et features est déjà défini pour vous.

Cette activité fait partie du cours

Introduction au Deep Learning avec PyTorch

Voir le cours

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# 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