Кількість класів

Щоб вирішити, чи використовувати бінарну чи багатокласову модель для задачі класифікації, потрібно знати кількість класів у даних. Набір даних Torch доступний у вашому робочому середовищі як train_dataset.

З огляду на кількість класів у train_dataset, яка задача класифікації зображень буде доречною?