1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorchで学ぶIntroduction to Deep Learning
Connected

演習

torchmetrics を使って精度を計算する

学習中に精度を追跡すると、最も性能の良いエポックを特定できます。

この演習では、3 クラスのマスク着用データセットに対して torchmetrics を使って精度を計算します。plot_errors 関数は誤分類されたサンプルを強調表示し、モデルのエラー分析に役立ちます。

torchmetrics パッケージはすでにインポート済みです。モデルの outputs は softmax の確率で、labels はワンホットベクトルです。

指示1 / 2

undefined XP
    1
    2
  • クラス数が3の "multiclass" 精度メトリクスを作成します。
  • データローダの各バッチに対して精度を計算します。