IniziaInizia gratis

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

Visualizza il corso

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
    ____
Modifica ed esegui il codice