Aan de slagGa gratis aan de slag

Een trainingslus schrijven

In scikit-learn is de trainingslus verpakt in de methode .fit(), terwijl je die in PyTorch handmatig opzet. Dat geeft flexibiliteit, maar vereist wel een eigen implementatie.

In deze oefening maak je een lus om een model te trainen voor salarisvoorspelling.

De functie show_results() is meegeleverd om een paar voorbeeldvoorspellingen te visualiseren.

De volgende pakketten zijn geïmporteerd: pandas als pd, torch, torch.nn als nn, torch.optim als optim, en DataLoader en TensorDataset uit torch.utils.data.

De volgende variabelen zijn aangemaakt: num_epochs, met het aantal epochs (op 5 gezet); dataloader, met de dataloader; model, met het neurale netwerk; criterion, met de verliesfunctie, nn.MSELoss(); optimizer, met de SGD-optimizer.

Deze oefening maakt deel uit van de cursus

Introductie tot Deep Learning met PyTorch

Cursus bekijken

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# Loop over the number of epochs and then the dataloader
for i in ____:
  for data in ____:
    # Set the gradients to zero
    ____
Code bewerken en uitvoeren