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
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 = ____