CommencezCommencez gratuitement

Évaluer des modèles de classification RNN

L'équipe de PyBooks souhaite maintenant que vous évaluiez le modèle RNN que vous avez créé et exécuté avec le jeu de données Newsgroup. Rappelez-vous, l'objectif était de classer les articles dans l'une des trois catégories :

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

Le modèle a été entraîné et vous avez affiché l'époque et la perte pour chaque modèle.

Utilisez torchmetrics pour évaluer différentes mesures pour votre modèle. Les éléments suivants ont été chargés pour vous : Accuracy, Precision, Recall, F1Score.

Une instance de rnn_model entraînée dans l'exercice précédent est aussi préchargée pour vous.

Cette activité fait partie du cours

Apprentissage profond pour le texte avec PyTorch

Voir le cours

Instructions de l’exercice

  • Créez une instance de chaque mesure pour la classification multiclasse avec num_classes égal au nombre de catégories.
  • Générez les prédictions pour rnn_model en utilisant les données de test X_test_seq.
  • Calculez les mesures à l'aide des classes prédites et des étiquettes réelles.

Exercice interactif pratique

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

# Create an instance of the metrics
accuracy = Accuracy(task="multiclass", ____)
precision = Precision(____, num_classes=____)
recall = Recall(task=____, num_classes=____)
f1 = F1Score(____, ____)

# Generate the predictions
outputs = ____(X_test_seq)
_, predicted = ____.____(outputs, 1)

# Calculate the metrics
accuracy_score = accuracy(____, y_test_seq)
precision_score = precision(____, y_test_seq)
recall_score = recall(____, y_test_seq)
f1_score = f1(____, y_test_seq)
print("RNN Model - Accuracy: {}, Precision: {}, Recall: {}, F1 Score: {}".format(accuracy_score, precision_score, recall_score, f1_score))
Modifier et exécuter le code