1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

Model pro vícetřídní klasifikaci

Máš-li k dispozici šablonu modelu pro binární klasifikaci, můžeš na jejím základě sestavit model pro víceřídní klasifikaci. Model by měl prostřednictvím parametru podporovat různé počty tříd – díky tomu ho v budoucnu snadno přizpůsobíš konkrétnímu úkolu.

Balíčky torch a torch.nn jako nn jsou již naimportované. Všechny obrázky mají rozměr 64×64 pixelů.

Pokyny

100 XP
  • Definuj metodu __init__ s parametry self a num_classes.
  • Vytvoř plně propojenou vrstvu se vstupní velikostí 16*32*32 a počtem výstupních tříd num_classes.
  • Vytvoř aktivační funkci softmax s parametrem dim=1.