search

rapidsai / cuml

cuML - RAPIDS Machine Learning Library
https://docs.rapids.ai/api/cuml/stable/
Apache License 2.0
4.25k stars 534 forks source link

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

Open CanOfSocks opened 1 month ago

CanOfSocks commented 1 month ago

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

dantegd commented 1 month ago

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: https://github.com/rapidsai/cuml/pull/6089

dantegd commented 1 month ago

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.