1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning for Text with PyTorch
Connected

Cvičení

Vyhodnocení klasifikačních modelů RNN

Tým PyBooks teď chce, abys vyhodnotil/a model RNN, který jsi vytvořil/a a spustil/a na datasetu Newsgroup. Cílem bylo klasifikovat články do jedné ze tří kategorií:

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

Model byl natrénován a ty jsi pro každou epochu vypsal/a její číslo a hodnotu ztrátové funkce.

Pomocí torchmetrics vyhodnoť různé metriky svého modelu. Pro tebe jsou již načteny tyto metriky: Accuracy, Precision, Recall, F1Score.

K dispozici máš také instanci rnn_model natrénovanou v předchozím cvičení.

Pokyny

100 XP
  • Vytvoř instanci každé metriky pro víceřídní klasifikaci a nastav num_classes na počet kategorií.
  • Vygeneruj predikce pro rnn_model pomocí testovacích dat X_test_seq.
  • Vypočítej metriky na základě predikovaných tříd a skutečných štítků.