1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch で学ぶ画像向け Deep Learning
Connected

演習

マルチクラス分類モデル

バイナリ分類モデルのひな型ができたので、これを発展させてマルチクラス分類モデルを設計しましょう。モデルはパラメータでクラス数を指定できるようにし、将来の特定のマルチクラス分類タスクに合わせて調整できるようにします。

torch と、torch.nn を nn としてインポート済みです。すべての画像サイズは 64x64 ピクセルです。

指示

100 XP
  • self と num_classes を引数に含む __init__ メソッドを定義します。
  • 入力サイズが 16*32*32、出力がクラス数 num_classes の全結合層を作成します。
  • dim=1 で活性化関数 softmax を作成します。