-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
56 lines (55 loc) · 2.01 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
import setuptools
setuptools.setup(
name='spider',
version='0.0.5',
author='Jason Corso, Laura Balzano and The University of Michigan DARPA D3M Spider Team',
author_email='jjcorso@umich.edu,girasole@umich.edu,davjoh@umich.edu,alsoltan@umich.edu,szetor@umich.edu',
url='https://github.com/dvdmjohnson/d3m_michigan_primitives',
license='MIT',
description='DARPA D3M Spider Project Code',
install_requires=[
'cvxpy (>=1.0.23)',
'd3m (==2020.1.9)',
'keras (==2.3.1)',
'librosa (==0.6.2)',
'matplotlib (>=1.5.1)',
'numpy (==1.17.3)',
'opencv-python (>=3.0.0)',
'pandas (==0.25.2)',
'Pillow (==6.2.1)',
'scikit-image (>=0.13.1)',
'scikit-learn (==0.21.3)',
'scipy (==1.3.1)',
'stopit (>=1.1.1)',
'tensorflow-gpu (==2.0.0)',
'torch (==1.3.1)',
'torchvision (==0.4.2)',
'typing (>=3.6.2)',
],
packages=[
'spider',
'spider.cluster',
'spider.cluster.ekss',
'spider.cluster.kss',
'spider.cluster.ssc_admm',
'spider.cluster.ssc_cvx',
'spider.cluster.ssc_omp',
'spider.pipelines',
'spider.supervised_learning',
'spider.supervised_learning.owl',
'spider.unsupervised_learning.grasta',
],
keywords='d3m_primitive',
entry_points = {
'd3m.primitives': [
'clustering.ekss.Umich = spider.cluster.ekss.ekss:EKSS',
'clustering.kss.Umich = spider.cluster.kss.kss:KSS',
'clustering.ssc_admm.Umich = spider.cluster.ssc_admm.ssc_admm:SSC_ADMM',
'clustering.ssc_cvx.Umich = spider.cluster.ssc_cvx.ssc_cvx:SSC_CVX',
'clustering.ssc_omp.Umich = spider.cluster.ssc_omp.ssc_omp:SSC_OMP',
'data_compression.grasta.Umich = spider.unsupervised_learning.grasta.grasta:GRASTA',
'regression.owl_regression.Umich = spider.supervised_learning.owl.owl:OWLRegression',
],
},
test_suite='spider.tests.suite'
)