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
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
____