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.
Toto cvičení je součástí kurzu
Introduction to Deep Learning with PyTorch
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
input_tensor = torch.tensor([[2.4]])
# Create a sigmoid function and apply it on input_tensor
sigmoid = nn.____()
probability = ____(____)
print(probability)