NLP, Python, Transformers, Data Science, Linear Regression.
function, the prediction is kind of confusing. According to this document from Hugging Face (line 140 to line 162), they will apply the sigmoid function over the output when the problem_type == “multi_label_classification” or num_labels == 1. My best explanation at this moment is when the