1. Learn
    2. /
  2. Courses
    3. /
  3. Apprentissage profond pour le texte avec PyTorch
Connected

Exercise

Créer un modèle LSTM pour le texte

Chez PyBooks, l'équipe cherche constamment à améliorer l'expérience des utilisateurs en tirant parti des plus récents progrès technologiques. Dans cette optique, on vous confie une tâche importante. L'équipe souhaite que vous exploriez le potentiel d'un autre outil puissant : LSTM, reconnu pour sa capacité à saisir des motifs de données plus complexes. Vous travaillez avec le même jeu de données Newsgroup, et l'objectif demeure le même : classifier les articles de nouvelles en trois catégories distinctes :

rec.autos, sci.med et comp.graphics.

Les modules suivants ont été chargés pour vous : torch, nn, optim.

Instructions

100 XP
  • Configurez un modèle LSTM en complétant les couches LSTM et linéaire avec les paramètres nécessaires.
  • Initialisez le modèle avec les paramètres requis.
  • Entraînez le modèle LSTM en réinitialisant les gradients à zéro et en faisant passer les données d'entrée X_train_seq dans le modèle.
  • Calculez la perte à partir des outputs prédits et des véritables étiquettes.