Skip to content

Adding scikit-learn-intelex package #1135

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

Merged
merged 1 commit into from
Mar 28, 2022
Merged

Conversation

napetrov
Copy link
Contributor

@napetrov napetrov commented Mar 6, 2022

This PR integrates accelerated x86 scikit-learn library that brings noticeable performance improvements for scikit-learn as well as set of AutoML frameworks(AutoGluon, autoscikit-learn, PyCaret, etc. ) that are using scikit-learn internally.

Link to github project - https://github.com/intel/scikit-learn-intelex . Accelerations achieved through use of oneDAL library - https://github.com/oneapi-src/oneDAL

There is set of users/notebooks that already use accelerations - https://www.kaggle.com/search?q=scikit-learn-intelex

Based on demonstrated results across notebooks we see that on Kaggle instances 2x-40x speedups can be achieved. This leads to productivity, results, CO2 reduction improvements.

@napetrov
Copy link
Contributor Author

@rosbo - appreciate if you can take a look on this request

@rosbo rosbo self-requested a review March 28, 2022 21:36
@rosbo rosbo self-assigned this Mar 28, 2022
@rosbo
Copy link
Contributor

rosbo commented Mar 28, 2022

Hi @napetrov,

The speedup numbers are impressive. Merging this PR. I will add tests to prevent regression in a separate PR.

Thank you

@rosbo rosbo merged commit f4d7262 into Kaggle:main Mar 28, 2022
rosbo added a commit that referenced this pull request Mar 28, 2022
sklearnex was added in #1135

http://227237481
rosbo added a commit that referenced this pull request Mar 29, 2022
sklearnex was added in #1135

http://227237481
rosbo added a commit that referenced this pull request Mar 29, 2022
sklearnex was added in #1135

http://227237481
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 this pull request may close these issues.

2 participants