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, ) +