diff --git a/requirements.txt b/requirements.txt
index 36aab79..31929da 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,6 +6,7 @@ mpmath
 numpy~=1.21.5
 pandas~=1.1.4
 scipy~=1.2.1
+sklearn~=1.0.2
 tensorflow-datasets>=4.4.0
 tensorflow-estimator>=2.3.0
 tensorflow-probability>=0.13.0
diff --git a/setup.py b/setup.py
index 49c4330..393b85f 100644
--- a/setup.py
+++ b/setup.py
@@ -34,6 +34,7 @@ setup(
         'numpy~=1.21.5',
         'pandas~=1.1.4',
         'scipy~=1.2.1',
+        'sklearn~=1.0.2',
         'tensorflow-datasets>=4.4.0',
         'tensorflow-estimator>=2.3.0',
         'tensorflow-probability>=0.13.0',