1. Learn
  2. /
  3. Kurser
  4. /
  5. Introducere în Deep Learning cu PyTorch

Connected

övning

Funcțiile sigmoid și softmax

Funcțiile sigmoid și softmax sunt funcții de activare esențiale în învățarea profundă, folosite adesea ca ultimul pas într-o rețea neuronală.

  • Sigmoid este folosită pentru clasificarea binară
  • Softmax este folosită pentru clasificarea multi-clasă

Pornind de la un tensor de ieșire pre-activare al unei rețele, aplică funcția de activare potrivită pentru a obține rezultatul final.

torch.nn a fost deja importat ca nn.

Instruktioner 1/2

undefined XP
  • 1
    • Creează o funcție sigmoid și aplică-o pe input_tensor pentru a genera o probabilitate într-o sarcină de clasificare binară.
  • 2
    • Creează o funcție softmax și aplică-o pe input_tensor pentru a genera o probabilitate într-o sarcină de clasificare multi-clasă.