1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie dla obrazów z PyTorch
Connected

ćwiczenie

Model klasyfikacji wieloklasowej

Mając gotowy szablon modelu klasyfikacji binarnej, możesz teraz rozbudować go o obsługę klasyfikacji wieloklasowej. Model powinien przyjmować liczbę klas jako parametr, dzięki czemu będzie można go łatwo dostosować do konkretnego zadania klasyfikacji wieloklasowej w przyszłości.

Biblioteki torch oraz torch.nn jako nn zostały już zaimportowane. Wszystkie obrazy mają rozmiar 64×64 piksele.

Instrukcje

100 XP
  • Zdefiniuj metodę __init__, przyjmując self i num_classes jako parametry.
  • Utwórz w pełni połączoną warstwę z rozmiarem wejściowym 16*32*32 i liczbą klas num_classes jako wyjściem.
  • Utwórz funkcję aktywacji softmax z parametrem dim=1.