CommencezCommencez gratuitement

Écrire une boucle d'entraînement

Dans scikit-learn, la boucle d'entraînement est encapsulée dans la méthode .fit(), tandis qu'avec PyTorch, on la met en place manuellement. Cela offre plus de flexibilité, mais exige une implémentation personnalisée.

Dans cet exercice, vous allez créer une boucle pour entraîner un modèle de prédiction de salaire.

La fonction show_results() est fournie pour vous aider à visualiser quelques prédictions d'exemple.

Les importations de modules fournies sont : pandas sous pd, torch, torch.nn sous nn, torch.optim sous optim, ainsi que DataLoader et TensorDataset de torch.utils.data.

Les variables suivantes ont été créées : num_epochs, qui contient le nombre d'époques (réglé à 5); dataloader, qui contient le chargeur de données; model, qui contient le réseau de neurones; criterion, qui contient la fonction de perte, nn.MSELoss(); optimizer, qui contient l'optimiseur SGD.

Cette activité fait partie du cours

Introduction au Deep Learning avec PyTorch

Voir le cours

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Loop over the number of epochs and then the dataloader
for i in ____:
  for data in ____:
    # Set the gradients to zero
    ____
Modifier et exécuter le code