クラス数を確認する

分類問題でバイナリモデルかマルチクラスモデルのどちらを使うかを判断するには、データ内のクラス数を把握する必要があります。Torch のデータセットはワークスペース内で train_dataset として利用できます。

train_dataset のクラス数を踏まえると、どの画像分類タスクが適切でしょうか。