Skip to content
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

[QST] Can cuML models be run with sklearn #6093

Open
CanOfSocks opened this issue Oct 2, 2024 · 2 comments
Open

[QST] Can cuML models be run with sklearn #6093

CanOfSocks opened this issue Oct 2, 2024 · 2 comments
Labels
? - Needs Triage Need team to review and classify cuml-cpu question Further information is requested

Comments

@CanOfSocks
Copy link

What is your question?
I have a Epsilon-Support Vector Regression (SVR) model that I'm using sklearn to fit. I'm currently deploying this in an environment that is only confirmed compatible with sklearn.

Is it possible to fit a model using cuML, but get predictions by loading it in a sklearn environment. A question was asked #4869, but I'm not sure if the CPU/GPU interoperability features have been implemented.

Thanks

@CanOfSocks CanOfSocks added ? - Needs Triage Need team to review and classify question Further information is requested labels Oct 2, 2024
@dantegd
Copy link
Member

dantegd commented Oct 4, 2024

Thanks for the issue @CanOfSocks! This is very good timing to ask, we are actively working right now (well @viclafargue ) to enable this to be possible for SVMs in the near future: #6089

@dantegd
Copy link
Member

dantegd commented Oct 8, 2024

Updating this issue, created #6102 to add an explicit and supported API to export (supported) cuML models to Scikit-learn. This issue will be solved when both of these PRs are merged.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
? - Needs Triage Need team to review and classify cuml-cpu question Further information is requested
Projects
None yet
Development

No branches or pull requests

3 participants