-
Notifications
You must be signed in to change notification settings - Fork 26.9k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
fix logit-to-multi-hot conversion in example #26936
fix logit-to-multi-hot conversion in example #26936
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks!
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should we not apply sigmoid ?
Otherwise fine with me. There is a way to pass a specific function to the trainer: preprocess_logits_for_metrics
which can apply the sigmoid otherwise (but not needed here)
@ArthurZucker I've updated the comment for educational purpose. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Cool thanks!
* fix logit to multi-hot converstion * add comments * typo
* fix logit to multi-hot converstion * add comments * typo
What does this PR do?
This PR fixes #26830 . The bug might lead to difference in measuring multi-label acc (Slightly higher precision and lower recall).
@younesbelkada Would you please review.