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

Add "Intel® Extension for Scikit-learn" for better performance #382

Open
Ankur-singh opened this issue Sep 29, 2022 · 3 comments · May be fixed by #384
Open

Add "Intel® Extension for Scikit-learn" for better performance #382

Ankur-singh opened this issue Sep 29, 2022 · 3 comments · May be fixed by #384

Comments

@Ankur-singh
Copy link
Contributor

Just by adding two lines of code, we can drastically improve the performance. Intel Extension for SciKit-learn package dynamically patches scikit-learn estimators while improving performance of machine learning algorithms particularly for Intel CPUs and GPUs.

More details regarding usage and benchmarks can be find here : https://www.intel.com/content/www/us/en/developer/tools/oneapi/scikit-learn.html

Just by adding these two lines at the top should do the job.

from sklearnex import patch_sklearn
patch_sklearn()

I can help if, anyone is interested in doing it. Else, I can do it.

Other contributors, let me know if you have any suggestions ?

@rahulunair
Copy link

@Ankur-singh i will take this up :), i always wanted a quick way to get initial preds, as lazypredict already does that, let's try to make it as fast as we can!

@Ankur-singh
Copy link
Contributor Author

sure @rahulunair. It would be great if you can also benchmark and share the results. Other members will also be able to see the performance gains because of Intel Extension for SciKit-learn.

rahulunair added a commit to rahulunair/lazypredict that referenced this issue Oct 2, 2022
Intel Scikit learn extensions is an sklearn api compatible library that
acclerates key compute intensive algorithms. For more info on perf
please see:https://github.com/intel/scikit-learn-intelex

Fixes: shankarpandala#382
@rahulunair
Copy link

@Ankur-singh just added a patch, tested with few toy datasets on an laptop with i7, got around ~1.5x perf improvement vs stock. If you have time, it would be nice to take one of the bigger datasets from this repo and compare perf as well: https://github.com/IntelPython/scikit-learn_bench

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants