-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
executable file
·28 lines (27 loc) · 1.12 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from setuptools import setup, find_packages
setup(
name='qshap',
version='0.3.5',
description='Exact computation of Shapley R-squared for tree ensembles in polynomial time',
long_description=open('README.md').read(),
long_description_content_type='text/markdown', # Specify that the long_description is in Markdown
author='Zhongli Jiang, Dabao Zhang',
author_email='[email protected], [email protected]',
license="GPL-2.0",
packages=find_packages(),
url="https://github.com/catstats/Q-SHAP",
setup_requires=['numpy'],
install_requires=['numpy', 'scikit-learn', 'shap', 'numba', 'ipywidgets', 'pandas', 'matplotlib'],
extras_require={'xgboost': ['xgboost'],'lightgbm': ['lightgbm']},
classifiers=[
'Operating System :: Microsoft :: Windows',
'Operating System :: Unix',
'Operating System :: MacOS',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
'Programming Language :: Python :: 3.11',
'Programming Language :: Python :: 3.12',
],
python_requires='>=3.7',
zip_safe=False
)