From e00ab2c66feec440bb01ac36d6bcf825f0cd07f7 Mon Sep 17 00:00:00 2001 From: perib Date: Tue, 20 Feb 2024 14:39:42 -0800 Subject: [PATCH] update compatibility with scikitlearn 1.4 --- requirements.txt | 2 +- setup.py | 2 +- tpot/gp_deap.py | 3 ++- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index d635f679..dc6a968a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ deap>=1.2 nose==1.3.7 numpy>=1.16.3 -scikit-learn>=1.1.3 +scikit-learn>=1.4.1 imbalanced-learn>=0.7.0 scipy>=1.3.1 tqdm>=4.36.1 diff --git a/setup.py b/setup.py index aca751ce..695ab8b1 100644 --- a/setup.py +++ b/setup.py @@ -37,7 +37,7 @@ def calculate_version(): zip_safe=True, install_requires=['numpy>=1.16.3', 'scipy>=1.3.1', - 'scikit-learn>=1.1.3', + 'scikit-learn>=1.4.1', 'deap>=1.2', 'update_checker>=0.16', 'tqdm>=4.36.1', diff --git a/tpot/gp_deap.py b/tpot/gp_deap.py index 4805d714..f509f67e 100644 --- a/tpot/gp_deap.py +++ b/tpot/gp_deap.py @@ -464,7 +464,8 @@ def _wrapped_cross_val_score(sklearn_pipeline, features, target, verbose=0, parameters=None, error_score='raise', - fit_params=sample_weight_dict) + fit_params=sample_weight_dict, + score_params=None) for train, test in cv_iter] if isinstance(scores[0], list): #scikit-learn <= 0.23.2 CV_score = np.array(scores)[:, 0]