diff --git a/Dockerfile b/Dockerfile
index 0d5ec4d..d31c3e4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,6 +10,7 @@ RUN apt install unzip -y
 RUN pip3 install kaggle
 RUN pip3 install pandas
 RUN pip3 install torch
+RUN pip3 install sklearn
 
 ARG CUTOFF
 ENV CUTOFF=${CUTOFF}