diff --git a/setup.py b/setup.py index 4094061..2cf353a 100644 --- a/setup.py +++ b/setup.py @@ -9,4 +9,7 @@ author_email='christoph.molnar@gmail.com', url='', packages=['rulefit'], + install_requires=['scikit-learn>=0.20.2', + 'numpy>=1.16.1', + 'pandas>=0.24.1'] )