1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch で学ぶテキストの Deep Learning
Connected

演習

RNN分類モデルの評価

PyBooksのチームは、Newsgroupデータセットを使って作成・実行したRNNモデルを評価してほしいと考えています。目的は、記事を次の3つのカテゴリのいずれかに分類することでした。

rec.autos、sci.med、comp.graphics

モデルはすでに学習済みで、各エポックの損失を出力しました。

torchmetricsを使って、モデルのさまざまな指標を評価してください。以下が読み込まれています：Accuracy、Precision、Recall、F1Score。

前の演習で学習したrnn_modelのインスタンスも事前に用意されています。

指示

100 XP
  • 複数クラス分類向けに、各メトリクスのインスタンスを作成し、num_classesはカテゴリ数に設定します。
  • テストデータX_test_seqを使って、rnn_modelの予測を生成します。
  • 予測クラスと正解ラベルを用いてメトリクスを計算します。