Skip to content

Blog

[PyTorch] Set the threshold of Sigmoid output and convert it to binary value

When using sigmoid function in PyTorch as our activation function, for example it is connected to the last layer of the model as the output of binary classification. After all, sigmoid can compress the value between 0-1, we only need to set a threshold, for example 0.5 and you can divide the value into two categories.

Read More »[PyTorch] Set the threshold of Sigmoid output and convert it to binary value