Experimentând cu dropout
Dropout ajută la prevenirea supraantrenării prin setarea aleatorie a unor valori de ieșire la zero în timpul antrenării. În acest exercițiu, vei construi o rețea neuronală simplă cu dropout și vei observa cum se comportă în modul de antrenare și în cel de evaluare.
Pachetul torch.nn este preîncărcat ca nn, iar features este deja definit pentru tine.
Acest exercițiu face parte din cursul
Introducere în Deep Learning cu PyTorch
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Model with Dropout
model = nn.Sequential(
nn.Linear(8, 6),
nn.Linear(6, 4),
____)
# Forward pass in training mode (Dropout active)
model.____
output_train = ____