remove a few regressors

Techtonique · Oct 9, 2024 · 45e2363 · 45e2363
1 parent 27df316
commit 45e2363
Show file tree

Hide file tree

Showing 9 changed files with 4 additions and 131 deletions.
diff --git a/examples/lazy_booster_classification.py b/examples/lazy_booster_classification.py
@@ -16,7 +16,7 @@
     X = data.data
     y= data.target
 
-    X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = .2, random_state = 123)
+    X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = .2, random_state = 13)
 
     clf = ms.LazyBoostingClassifier(verbose=0, ignore_warnings=True, #n_jobs=2,
                                     custom_metric=None, preprocess=False)

diff --git a/mlsauce.egg-info/PKG-INFO b/mlsauce.egg-info/PKG-INFO
diff --git a/mlsauce.egg-info/SOURCES.txt b/mlsauce.egg-info/SOURCES.txt
diff --git a/mlsauce.egg-info/dependency_links.txt b/mlsauce.egg-info/dependency_links.txt
diff --git a/mlsauce.egg-info/not-zip-safe b/mlsauce.egg-info/not-zip-safe
diff --git a/mlsauce.egg-info/requires.txt b/mlsauce.egg-info/requires.txt
diff --git a/mlsauce.egg-info/top_level.txt b/mlsauce.egg-info/top_level.txt
diff --git a/mlsauce/lazybooster/config.py b/mlsauce/lazybooster/config.py
@@ -17,7 +17,7 @@
     "HistGradientBoostingRegressor",
     "IsotonicRegression",
     "MLPRegressor",
-    # "KernelRidge",
+    "KernelRidge",
    #"MultiOutputRegressor",
     #"MultiTaskElasticNet",
     "MultiTaskElasticNetCV",
@@ -58,6 +58,7 @@
         and (est[0] not in removed_regressors)
         and (est[0][:5] != "Multi")
         and (est[0][-2:] != "CV")
+        and (est[0] not in ("SVR", "HuberRegressor", "LassoLarsIC"))
         and is_multitask_estimator(est[1]()) == False
     )
 ]
diff --git a/setup.py b/setup.py
@@ -37,7 +37,7 @@
 MAINTAINER_EMAIL = '[email protected]'
 LICENSE = 'BSD3 Clause Clear'
 
-__version__ = '0.22.1'
+__version__ = '0.22.2'
 
 VERSION = __version__