diff --git a/README.md b/README.md
index e168ba9fdc63f1b476c4af05c205480b8de1f6d0..9af2e50bb0741e6230b498c7deae72bb2b28f7e6 100644
--- a/README.md
+++ b/README.md
@@ -24,7 +24,7 @@ source env/bin/activate
 pip install --force-reinstall --index-url https://pypi.anaconda.org/intel/simple --no-dependencies numpy scipy==1.7.3
 pip install numpy scipy==1.7.3
 pip install torch --index-url https://download.pytorch.org/whl/cpu
-pip install joblib
+pip install joblib scikit-learn
 
 # install dependencies for the analysis in the notebook
 pip install matplotlib lmfit seaborn extra_data
@@ -50,7 +50,7 @@ pip install --force-reinstall --index-url https://pypi.anaconda.org/intel/simple
 pip install numpy scipy==1.7.3
 # install PyTorch
 pip install torch --index-url https://download.pytorch.org/whl/cpu
-pip install joblib
+pip install joblib scikit-learn
 ```
 
 ## Usage