From 21af4c4acc6b75d2a08a2a369ddefd5c8367daac Mon Sep 17 00:00:00 2001 From: nickotto Date: Fri, 28 Jul 2023 12:11:51 -0700 Subject: [PATCH] regressor time update for unit test --- tpot2/tests/test_estimators.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tpot2/tests/test_estimators.py b/tpot2/tests/test_estimators.py index 8a46b74a..d3a771e0 100644 --- a/tpot2/tests/test_estimators.py +++ b/tpot2/tests/test_estimators.py @@ -8,7 +8,6 @@ @pytest.fixture def tpot_estimator(): return tpot2.TPOTEstimator( population_size=10, - generations=2, scorers=['roc_auc_ovr'], scorers_weights=[1], classification=True, @@ -86,7 +85,7 @@ def tpot_classifier(): @pytest.fixture def tpot_regressor(): - return tpot2.tpot_estimator.templates.TPOTRegressor(max_time_seconds=30,verbose=1) + return tpot2.tpot_estimator.templates.TPOTRegressor(max_time_seconds=60,verbose=1) def test_tpot_classifier_fit(tpot_classifier,sample_dataset): #load iris dataset