Menulis training loop
Di scikit-learn, training loop dibungkus dalam metode .fit(), sedangkan di PyTorch, loop disusun secara manual. Ini memberikan fleksibilitas, tetapi memerlukan implementasi khusus.
Pada latihan ini, Anda akan membuat loop untuk melatih model prediksi gaji.
Fungsi show_results() disediakan untuk membantu Anda memvisualisasikan beberapa prediksi contoh.
Impor paket yang disediakan adalah: pandas sebagai pd, torch, torch.nn sebagai nn, torch.optim sebagai optim, serta DataLoader dan TensorDataset dari torch.utils.data.
Variabel berikut telah dibuat: num_epochs, berisi jumlah epoch (disetel ke 5); dataloader, berisi dataloader; model, berisi neural network; criterion, berisi fungsi loss, nn.MSELoss(); optimizer, berisi optimizer SGD.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Loop over the number of epochs and then the dataloader
for i in ____:
for data in ____:
# Set the gradients to zero
____