From 855156497cab9dc9e4066f6d22447fce07f9941e Mon Sep 17 00:00:00 2001 From: Saurabh Rawat Date: Mon, 4 Mar 2024 20:44:00 +0530 Subject: [PATCH] use this command pip install scikit-learn==0.24.0 pip install scipy==1.6.3 --- lazypredict/Supervised.py | 5 +++++ setup.py | 1 + 2 files changed, 6 insertions(+) diff --git a/lazypredict/Supervised.py b/lazypredict/Supervised.py index bf61208..949e835 100644 --- a/lazypredict/Supervised.py +++ b/lazypredict/Supervised.py @@ -13,6 +13,9 @@ from sklearn.preprocessing import StandardScaler, OneHotEncoder, OrdinalEncoder from sklearn.compose import ColumnTransformer from sklearn.utils import all_estimators +from lazypredict.Supervised import LazyRegressor + + from sklearn.base import RegressorMixin from sklearn.base import ClassifierMixin from sklearn.metrics import ( @@ -48,6 +51,7 @@ "OutputCodeClassifier", "RadiusNeighborsClassifier", "VotingClassifier", + "SVC", # Add SVC to the list of removed classifiers ] removed_regressors = [ @@ -66,6 +70,7 @@ "RadiusNeighborsRegressor", "RegressorChain", "VotingRegressor", + "Ridge", # Add Ridge to the list of removed regressors ] CLASSIFIERS = [ diff --git a/setup.py b/setup.py index 5d740fa..c998f62 100644 --- a/setup.py +++ b/setup.py @@ -55,3 +55,4 @@ version='0.2.12', zip_safe=False, ) +