Skip to content

Machine Learning

[Machine Learning] Introduction the indicators of the three evaluation models of Precision、Recall、F1-score

Precision, Recall, and F1-score are three fairly well-known model evaluation indicators, which are mostly used for binary classification (if it is a multi-classification, it is suitable for macro and micro). The following is a brief description of these different indicators:

Read More »[Machine Learning] Introduction the indicators of the three evaluation models of Precision、Recall、F1-score

[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