diff --git a/setup.py b/setup.py index 2d83ca0..964aec4 100644 --- a/setup.py +++ b/setup.py @@ -94,7 +94,7 @@ def run(self): 'pandas==0.25.3', 'seaborn==0.9.0', 'matplotlib==3.1.1', - 'scikit-learn==0.21.2', + 'scikit-learn==1.5.0', 'numpy==1.22.0', 'imbalanced-learn==0.5.0', ],