-
Notifications
You must be signed in to change notification settings - Fork 527
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] MNMG KNN Classifier & Regressor #2211
Conversation
Please update the changelog in order to start CI tests. View the gpuCI docs here. |
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.
This is a great start. I think there's a lot of opportunity for reuse here and following the class hierarchy of the single-GPU implementations should be a big help there. I think it's a good idea to knock these out in this PR as it will make the kneighbors regressor much easier to build.
Seems like there was a cuDF update that causes some tests to fail. |
rerun tests |
1 similar comment
rerun tests |
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! It's great seeing all this code outside of cumlprims.
Was linked to cumlprims MR #37, corresponding code has been transfered in this PR.