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] Support and enable convert_dtype in estimator predict #2723
[REVIEW] Support and enable convert_dtype in estimator predict #2723
Conversation
Please update the changelog in order to start CI tests. View the gpuCI docs here. |
Discussed with @dantegd offline. Moving this back to WIP to rework and enable/allow |
…_score; pass kwargs for regressormixin score to predict like in classifiermixin; update ridge and elasticnet predict to use the convert_dtype
Moving back to WIP for some small updates |
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.
Few minor question, otherwise it looks good!
The failing linear regression test isn't failing locally. Triaging |
This PR:
convert_dtype
duringfit
and some of the existing ones that convert during predict.convert_dtype
parameter in logistic regression predictThere may be a bit more to complexity to convert_dtype for SVC(probability=True), so I plan to address it in a follow-up to streamline this PR.
This closes #2716