forked from Trusted-AI/AIX360
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
39 lines (36 loc) · 1.07 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
29
30
31
32
33
34
35
36
37
38
39
import setuptools
version = '0.1'
with open("aix360/version.py", 'w') as f:
f.write('# generated by setup.py\nversion = "{}"\n'.format(version))
setuptools.setup(
name='AIX360',
version=version,
description='IBM AI Explainability 360',
authos='aix360 developers',
url='https://github.com/IBM/AIX360',
author_email='aix360@us.ibm.com',
packages=setuptools.find_packages(),
license='Apache License 2.0',
long_description=open("README.md", 'r').read(),
install_requires=[
'joblib>=0.11',
'scikit-learn>=0.21.2',
'torch',
'torchvision',
'cvxpy',
'cvxopt',
'Image',
'keras',
'matplotlib',
'numpy',
'pandas',
'scipy>=0.17',
'tensorflow',
'xport',
'scikit-image',
'requests'
],
package_data={'aix360': ['data/*', 'data/*/*', 'data/*/*/*', 'models/*', 'models/*/*', 'models/*/*/*']},
include_package_data=True,
zip_safe=False
)