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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
- from os import path
- import setuptools
- path_to_repo = path.abspath(path.dirname(__file__))
- with open(path.join(path_to_repo, 'readme.md'), encoding='utf-8') as f:
- long_description = f.read()
- required_pypi = [
- 'matplotlib',
- 'mlxtend>=0.18.0', # some lower versions are missing fpgrowth
- 'numpy',
- 'pandas<=2.1.4', # pandas 2.2 introduced some issues with the query function
- 'requests', # used in c4.5
- 'scipy',
- 'scikit-learn>=1.2.0', # recently updated this
- 'tqdm', # used in BART
- ]
- extra_deps = [
- 'cvxpy', # optionally requires cvxpy for slim
- 'corels', # optionally requires corels for optimalrulelistclassifier
- 'gosdt-deprecated', # optionally requires gosdt for optimaltreeclassifier
- 'irf', # optionally require irf for iterativeRandomForestClassifier
- ]
- setuptools.setup(
- name="imodels",
- version="1.4.2",
- author="Chandan Singh, Keyan Nasseri, Matthew Epland, Yan Shuo Tan, Omer Ronen, Tiffany Tang, Abhineet Agarwal, Theo Saarinen, Bin Yu, and others",
- author_email="chandan_singh@berkeley.edu",
- description="Implementations of various interpretable models",
- long_description=long_description,
- long_description_content_type="text/markdown",
- url="https://github.com/csinva/imodels",
- packages=setuptools.find_packages(
- exclude=['tests', 'tests.*', '*.test.*']
- ),
- install_requires=required_pypi,
- extras_require={
- 'dev': [
- 'dvu',
- 'gdown',
- # 'irf',
- 'jupyter',
- 'jupytext',
- 'matplotlib',
- # 'pdoc3', # for building docs
- 'pytest',
- 'pytest-cov',
- # 'seaborn', # in bartpy.diagnostics.features
- 'slurmpy',
- # 'statsmodels', # in bartpy.diagnostics.diagnostics
- # 'torch', # for neural-net-integrated models
- 'tqdm',
- 'pmlb',
- ]
- },
- python_requires='>=3.9.0',
- classifiers=[
- "Programming Language :: Python :: 3",
- "License :: OSI Approved :: MIT License",
- "Operating System :: OS Independent",
- ],
- )
|