diff --git a/Dockerfile b/Dockerfile
index 0416f5b..e1d4373 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,6 +12,7 @@ RUN pip3 install jovian
 RUN pip3 install torch
 RUN pip3 install seaborn
 RUN pip3 install torchvision
+RUN pip3 install sklearn
 # RUN python3 -m pip install kaggle
 RUN python3 -m pip install pandas
 RUN pip3 install matplotlib