-
Notifications
You must be signed in to change notification settings - Fork 29
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
Initial sklearnex support #102
Conversation
Thanks for the contribution! I did a quick test on my machine, and it works correctly. Some thoughts:
|
params_ElasticNet throws an error because log=True requires the search space to be >0. The solution could be to either:
or
|
Realized I could simply make the commit. I think this is functional now and should be ready to merge |
Thanks for the review and merge @perib @nickotto - a couple other things we were thinking:
Please let me know if any of these additions would be of interest. |
[please review the Contribution Guidelines prior to submitting your pull request. go ahead and delete this line if you've already reviewed said guidelines.]
What does this PR do?
Adds configs for and enables use of the Intel(R) Extension for Scikit-learn in tpot2.
How should this PR be tested?
Ensure backwards compatibility - make sure previous configurations run without issue.
Test usage of the config files both using TPOTEstimator and TPOTRegressor/TPOTClassifier with root_config_dict set to classifiers_sklearnex or regressors_sklearnex respectively.
Any background context you want to provide?
https://github.com/intel/scikit-learn-intelex/
Notebook with demonstration of speedup capabilities in tpot: https://colab.research.google.com/drive/1yeqbKg2Sv8affoHrOCsluunscjQnmqy5#scrollTo=h_TKDu-FnAFZ&line=5&uniqifier=1
What are the relevant issues?
#1316
Questions: