diff --git a/setup.py b/setup.py
index 837c925..3673be4 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
       install_requires=[
           'numpy',
           'scipy',
-          'scikit-learn==0.20',
+          'scikit-learn==1.5.0',
           'torch',
           'matplotlib',
           'tensorboardX',