From d55552c974ca1b6220d8ac04278128b4951d0d23 Mon Sep 17 00:00:00 2001 From: rahulunair Date: Sun, 2 Oct 2022 16:41:35 -0700 Subject: [PATCH] (perf): enabling sklearn extensions to improve performance 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: #382 --- lazypredict/Supervised.py | 2 ++ requirements.txt | 3 ++- requirements_dev.txt | 3 ++- 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/lazypredict/Supervised.py b/lazypredict/Supervised.py index bf61208..ff6ec7b 100644 --- a/lazypredict/Supervised.py +++ b/lazypredict/Supervised.py @@ -8,6 +8,8 @@ from tqdm import tqdm import datetime import time +from sklearnex import patch_sklearn +patch_sklearn() from sklearn.pipeline import Pipeline from sklearn.impute import SimpleImputer, MissingIndicator from sklearn.preprocessing import StandardScaler, OneHotEncoder, OrdinalEncoder diff --git a/requirements.txt b/requirements.txt index 37ca346..f15fc95 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,8 @@ click scikit-learn +scikit-learn-intelex pandas tqdm joblib lightgbm -xgboost \ No newline at end of file +xgboost diff --git a/requirements_dev.txt b/requirements_dev.txt index 45bd550..551e1f5 100644 --- a/requirements_dev.txt +++ b/requirements_dev.txt @@ -1,9 +1,10 @@ click scikit-learn +scikit-learn-intelex pandas tqdm joblib lightgbm xgboost pytest -myst-parser \ No newline at end of file +myst-parser