1. Apprendre
    2. /
  2. Cours
    3. /
  3. Apprentissage profond intermédiaire avec PyTorch
Connected

Exercice

Évaluer des modèles de prévision

C'est le moment d'évaluer ! Le même réseau LSTM que vous avez entraîné dans l'exercice précédent a été entraîné pour vous pendant quelques époques de plus et est disponible sous le nom net.

Votre tâche est de l'évaluer sur un jeu de test à l'aide de la mesure de l'erreur quadratique moyenne (la bibliothèque torchmetrics a déjà été importée pour vous). Voyons comment le modèle s'en sort !

Instructions

100 XP
  • Définissez la mesure de l'erreur quadratique moyenne et assignez-la à mse.
  • Passez la séquence d'entrée à net, puis appliquez squeeze au résultat avant de l'assigner à outputs.
  • Calculez la valeur finale de la mesure de test et assignez-la à test_mse.