diff --git a/requirements.txt b/requirements.txt index 43624ce..e3ee3f7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,6 +3,7 @@ numpy>=1.14.0,<2.0.0 oauth2client<4.0.0 pandas>=0.22.0,<1.0.0 six>=1.11.0,<2.0.0 +scikit-learn>=0.18,<0.21 tensorflow==1.13.1 tensorflow-transform==0.13.0 tensorflow-data-validation==0.13.1