Experimentar con el abandono
El abandono ayuda a evitar el sobreajuste, poniendo aleatoriamente a cero algunos valores de salida durante el entrenamiento. En este ejercicio, construirás una red neuronal sencilla con abandono y observarás cómo se comporta en los modos de entrenamiento y evaluación.
torch.nn
está precargado como nn
, y features
ya está definido para ti.
Este ejercicio forma parte del curso
Introducción al aprendizaje profundo con PyTorch
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
# Model with Dropout
model = nn.Sequential(
nn.Linear(8, 6),
nn.Linear(6, 4),
____)
# Forward pass in training mode (Dropout active)
model.____
output_train = ____