add multitask regressors to Lazy*Classifier Pt.2

Techtonique · Oct 9, 2024 · afe21c4 · afe21c4
1 parent ddc5350
commit afe21c4
Show file tree

Hide file tree

Showing 4 changed files with 481 additions and 242 deletions.
diff --git a/examples/lazy_booster_classification.py b/examples/lazy_booster_classification.py
@@ -18,7 +18,7 @@
 
     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, 
+    clf = ms.LazyBoostingClassifier(verbose=0, ignore_warnings=True, #n_jobs=2,
                                     custom_metric=None, preprocess=False)
 
     start = time()

diff --git a/examples/lazy_booster_regression.py b/examples/lazy_booster_regression.py
@@ -11,7 +11,7 @@
 y= data.target
 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = .2, random_state = 123)
 
-regr = ms.LazyBoostingRegressor(verbose=0, ignore_warnings=True, 
+regr = ms.LazyBoostingRegressor(verbose=0, ignore_warnings=True, n_jobs=2,
                                 custom_metric=None, preprocess=True)
 models, predictioms = regr.fit(X_train, X_test, y_train, y_test)
 model_dictionary = regr.provide_models(X_train, X_test, y_train, y_test)