Update instructions.

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