RNN-Klassifikationsmodelle bewerten
Das Team von PyBooks möchte nun, dass du das von dir erstellte und mit dem Newsgroup-Datensatz ausgeführte RNN-Modell bewertest. Zur Erinnerung: Das Ziel war, die Artikel einer von drei Kategorien zuzuordnen:
rec.autos, sci.med und comp.graphics.
Das Modell wurde trainiert und du hast für jede Epoche den Loss ausgegeben.
Verwende torchmetrics, um verschiedene Metriken für dein Modell zu berechnen. Folgendes wurde für dich geladen: Accuracy, Precision, Recall, F1Score.
Eine Instanz von rnn_model, die in der vorherigen Übung trainiert wurde, ist ebenfalls für dich vorab geladen.
Diese Übung ist Teil des Kurses
Deep Learning für Text mit PyTorch
Anleitung zur Übung
- Erstelle für jede Metrik eine Instanz für Mehrklassenklassifikation, mit
num_classesgleich der Anzahl der Kategorien. - Erzeuge die Vorhersagen für das
rnn_modelmit den TestdatenX_test_seq. - Berechne die Metriken mit den vorhergesagten Klassen und den echten Labels.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# 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))