diff --git a/setup.py b/setup.py index 5f4bd30..fbcc14c 100755 --- a/setup.py +++ b/setup.py @@ -26,7 +26,7 @@ 'pandas', 'statsmodels>=0.11.0', 'scipy', - 'sklearn', + 'scikit-learn', 'tensorflow'] if __name__ == "__main__":