-
-
Notifications
You must be signed in to change notification settings - Fork 26.1k
[MRG] BUG: enforce row-wise arg max of decision function=predicted class #15504
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
Conversation
The changes look good to me. Is there a test associated with this already? If not, you can take the code from the issue you reference, turn it into a test, make sure it fails before this change, and make sure it passes after the change. |
I believe there is already a test set up here: https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/tests/test_multiclass.py#L218 |
Did it fail before this fix? (Ideally you want to write a test that fails before your fix, and then passes after your fix) |
0841f0e
to
297d352
Compare
I added a small test that fails without the fix. |
…e predicted class
23733bf
to
05bd8b6
Compare
this is still waiting for reviews, right? |
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.
LGTM. Thanks
Reference Issues/PRs
Fixes #14124
What does this implement/fix? Explain your changes.
The estimators are traversed in reversed order to be consistent with the value returned on arg-max of decision function.
Any other comments?
The reproducer in the original issue, now runs successfully.