Ajuste do dropout
O dropout ajuda a evitar o sobreajuste definindo aleatoriamente alguns valores de saída como zero durante o treinamento. Neste exercício, você deve criar uma rede neural simples com dropout e observar como ela se comporta nos modos de treinamento e avaliação.
torch.nn
foi pré-carregado como nn
, e as variáveis independentes (features
) já foram definidas para você.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Model with Dropout
model = nn.Sequential(
nn.Linear(8, 6),
nn.Linear(6, 4),
____)
# Forward pass in training mode (Dropout active)
model.____
output_train = ____