ComeçarComece de graça

As funções sigmoide e softmax

As funções sigmoide e softmax são funções de ativação essenciais no aprendizado profundo, geralmente usadas como etapa final de uma rede neural.

  • A sigmoide serve para classificação binária
  • A softmax serve para classificação multiclasse

Dado um tensor de saída de pré-ativação de uma rede, aplique a função de ativação apropriada para obter a saída final.

torch.nn já foi importado como nn.

Este exercício faz parte do curso

Introdução ao Aprendizado Profundo com o PyTorch

Ver curso

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

input_tensor = torch.tensor([[2.4]])

# Create a sigmoid function and apply it on input_tensor
sigmoid = nn.____()
probability = ____(____)
print(probability)
Editar e executar o código