1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Introduction to Deep Learning with PyTorch
Connected

cvičení

Funkce sigmoid a softmax

Funkce sigmoid a softmax jsou klíčové aktivační funkce v deep learningu, které se často používají jako poslední krok v neuronové síti.

  • Sigmoid slouží pro binární klasifikaci
  • Softmax slouží pro klasifikaci do více tříd

Máš k dispozici tensor s výstupem sítě před aktivací – aplikuj na něj odpovídající aktivační funkci a získej finální výstup.

torch.nn už je naimportován jako nn.

Instrukce 1/2

undefined XP
  • 1
    • Vytvoř funkci sigmoid a aplikuj ji na input_tensor, abys získal/a pravděpodobnost pro úlohu binární klasifikace.
  • 2
    • Vytvoř funkci softmax a aplikuj ji na input_tensor, abys získal/a pravděpodobnost pro úlohu klasifikace do více tříd.