Skip to content

Fix multi-label prediction thresholding in run_classification#43135

Closed
sumukhacharya03 wants to merge 2 commits intohuggingface:mainfrom
sumukhacharya03:issue-43116-multilabel-fix
Closed

Fix multi-label prediction thresholding in run_classification#43135
sumukhacharya03 wants to merge 2 commits intohuggingface:mainfrom
sumukhacharya03:issue-43116-multilabel-fix

Conversation

@sumukhacharya03
Copy link
Copy Markdown

Fixes incorrect multi-label prediction thresholding that caused empty
predictions in predict_results.txt for multi-label classification.

@Rocketknight1
Copy link
Copy Markdown
Member

expit is the sigmoid function and sigmoid(0) = 0.5, so expit(p) > 0.5 and p > 0 are the same boolean tensor. I'm not sure what this changes!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants