-
Notifications
You must be signed in to change notification settings - Fork 17
/
setup.py
76 lines (71 loc) · 2.91 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
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
66
67
68
69
70
71
72
73
74
75
76
from setuptools import setup, find_packages
print(find_packages())
long_description = """
DeepArchitect is an architecture search framework with a focus on modularity,
extensibility, composability, and ease of use.
DeepArchitect uses composable and modular operators to express search
spaces over computational graphs that are then passed to search algorithms that
sample architectures from them with the goal of maximizing a desired performance
metric.
We aim to impact the workflows of researchers and practitioners with DeepArchitect.
For researchers, DeepArchitect aims to make architecture search research more
reusable and reproducible by providing them with a modular framework that they
can use to implement new search algorithms and new search spaces while reusing
a large amount of existing code.
For practicioners, DeepArchitect aims to augment their workflow by providing them
with a tool that allows them to easily write a search space encoding the large
number of choices involved in designing an architecture and use a search
algorithm automatically find an architecture in the search space.
DeepArchitect has the following **main components**:
* a language for writing composable and expressive search spaces over computational
graphs in arbitrary domains (e.g., Tensorflow, Keras, Pytorch, and even
non deep learning frameworks such as scikit-learn and preprocessing pipelines);
* search algorithms that can be used for arbitrary search spaces;
* logging functionality to easily keep track of the results of a search;
* visualization functionality to explore and inspect logging information resulting
from a search experiment.
"""
setup(
name='deep_architect',
version='0.1.0',
description=
"DeepArchitect: Architecture search so easy that you'll think it's magic!",
long_description=long_description,
url='https://github.com/negrinho/deep_architect',
long_description_content_type='text/markdown',
keywords=[
'architecture search',
'framework',
'deep learning',
'pytorch',
'tensorflow',
],
license='MIT',
author='Renato Negrinho',
author_email='[email protected]',
# See https://pypi.python.org/pypi?%3Aaction=list_classifiers
classifiers=[
'Development Status :: 3 - Alpha',
'Intended Audience :: Science/Research',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7'
],
packages=find_packages(include=["deep_architect*"]),
python_requires=">=3.6",
install_requires=[
'numpy',
'scipy',
"scikit-learn",
"tensorflow==1.15",
"torch>=1.2",
"keras>=2.3",
"matplotlib",
"graphviz"
],
extras_require={
"docs": ["sphinx"
"sphinx_rtd_theme"],
"viz": ['matplotlib', 'graphviz'],
"explorer": ["dash==1.0.1", "dash-daq==0.1.0"]
})