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