嘗試使用 dropout
Dropout 透過在訓練期間隨機將部分輸出設為 0，來幫助防止過度擬合。這個練習中，你會建立一個含有 dropout 的簡單神經網路，並觀察它在訓練與評估模式下的行為差異。
torch.nn 套件已預先載入為 nn，而 features 也已為你定義好。
本練習屬於課程
使用 PyTorch 的深度學習入門
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
# Model with Dropout
model = nn.Sequential(
nn.Linear(8, 6),
nn.Linear(6, 4),
____)
# Forward pass in training mode (Dropout active)
model.____
output_train = ____