diff --git a/requirements.txt b/requirements.txt
index b6b14ff..6257780 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,5 +2,5 @@ matplotlib==2.2.2
 torch==1.8.1
 pandas==0.22.0
 numpy==1.14.3
-scikit_learn==0.24.0
+scikit_learn==1.0.1
 torchvision==0.9.1