Scrivere un training loop
In scikit-learn, il training loop è incapsulato nel metodo .fit(), mentre in PyTorch va impostato manualmente. Questo offre più flessibilità, ma richiede un'implementazione personalizzata.
In questo esercizio, creerai un loop per addestrare un modello di previsione degli stipendi.
La funzione show_results() è fornita per aiutarti a visualizzare alcune previsioni di esempio.
Sono già stati importati i seguenti pacchetti: pandas come pd, torch, torch.nn come nn, torch.optim come optim, oltre a DataLoader e TensorDataset da torch.utils.data.
Sono state create le seguenti variabili: num_epochs, che contiene il numero di epoche (impostato a 5); dataloader, che contiene il dataloader; model, che contiene la rete neurale; criterion, che contiene la funzione di perdita, nn.MSELoss(); optimizer, che contiene l'ottimizzatore SGD.
Questo esercizio fa parte del corso
Introduzione al Deep Learning con PyTorch
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# Loop over the number of epochs and then the dataloader
for i in ____:
for data in ____:
# Set the gradients to zero
____