開始使用免費開始

撰寫訓練迴圈

scikit-learn 中,訓練迴圈包在 .fit() 方法裡;而在 PyTorch,則需要手動設定。這雖然更有彈性,但也需要你自行實作。

在這個練習中,你將撰寫一個迴圈來訓練薪資預測模型。

已提供 show_results() 函式,協助你視覺化部分樣本的預測結果。

已匯入的套件如下:pandas 作為 pdtorchtorch.nn 作為 nntorch.optim 作為 optim,以及從 torch.utils.data 匯入的 DataLoaderTensorDataset

已建立的變數如下:num_epochs(訓練回合數,設為 5)、dataloader(資料載入器)、model(神經網路)、criterion(損失函式,nn.MSELoss())、optimizer(SGD 最佳化器)。

本練習屬於課程

使用 PyTorch 的深度學習入門

檢視課程

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Loop over the number of epochs and then the dataloader
for i in ____:
  for data in ____:
    # Set the gradients to zero
    ____
編輯並執行程式碼