-
Notifications
You must be signed in to change notification settings - Fork 532
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
[REVIEW] Add cython hinge_loss #3409
[REVIEW] Add cython hinge_loss #3409
Conversation
Looks like this was hit by the IVFPQ error on CUDA 11 and network errors on 10.1 and 10.2. Going to rerun. |
rerun tests |
@Nanthini10 Could you merge branch-0.18 into your branch? We just got the xfail on IVFPQ in, and I think that's the only thing keeping your PR from passing. The other failures were timeouts on the clone. |
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.
One small tweak for a redundant check, and one question (which I'm honestly not sure the answer to), but otherwise this looks great!
I know you mentioned it to me offline, but would you mind also adding a comment (either on the PR or in the code) that explains why we're not using the hinge_loss implemented in the CUDA/CPP layer? I figure having that there may help avoid confusion down the line about why this doesn't match the pattern of other metrics.
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.
Everything looks good to me. Well done.
Should this be targeting 0.19? |
Yes, this is for 0.19 @beckernick |
@Nanthini10 I changed the base to |
Codecov Report
@@ Coverage Diff @@
## branch-0.19 #3409 +/- ##
===============================================
+ Coverage 71.77% 79.35% +7.57%
===============================================
Files 212 226 +14
Lines 17075 18007 +932
===============================================
+ Hits 12256 14289 +2033
+ Misses 4819 3718 -1101
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
@gpucibot merge |
Implements hinge_loss with tests. API matches and code is adapted from sklearn https://scikit-learn.org/stable/modules/generated/sklearn.metrics.hinge_loss.html
NOTE: We're not using the C++ hinge implementation in order to comply with sklearn API. The C++ implementation is used in the intermediate step in SGD and accept different set of parameters than what is used to calculate hinge_loss.