-
Notifications
You must be signed in to change notification settings - Fork 4
/
setup.py
40 lines (39 loc) · 1.06 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
40
from setuptools import setup, find_packages
setup(
name = 'EA-HAS-Bench',
version = '1.0',
author = 'Shuguang Dou, Xinyang Jiang, Cairong Zhao, Dongshen Li',
author_email = 'xinyangjiang@microsoft.com',
description = 'Energy-aware HAS Benchmark',
license = 'MIT License',
keywords = ['AutoML', 'NAS', 'Deep Learning'],
url = "https://github.com/microsoft/EA-HAS-Bench",
packages=find_packages(),
include_package_data=True,
classifiers=[
"Programming Language :: Python :: 3",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
install_requires = [
'autograd>=1.3',
'click',
'Cython',
'ConfigSpace==0.4.12',
'ipython',
'lightgbm>=2.3.1',
'matplotlib',
'numpy==1.22.0',
'pandas',
'pathvalidate',
'Pillow>=7.1.2',
'psutil',
'scikit-image',
'scikit-learn>=0.23.1',
'seaborn==0.11.1',
'statsmodels',
'tqdm',
'xgboost',
'fvcore',
'hyperopt',
]
)