-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
72 lines (66 loc) · 3.47 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
from setuptools import setup
from setuptools.command.develop import develop
from setuptools.command.install import install
class PostDevelopCommand(develop):
"""Post-installation for development mode."""
def run(self):
import keras.applications.resnet50 as resnet50
import keras.applications.vgg16 as vgg16
resnet50.ResNet50(weights='imagenet')
vgg16.VGG16(weights='imagenet', include_top=False)
develop.run(self)
class PostInstallCommand(install):
"""Post-installation for installation mode."""
def run(self):
import keras.applications.resnet50 as resnet50
import keras.applications.vgg16 as vgg16
resnet50.ResNet50(weights='imagenet')
vgg16.VGG16(weights='imagenet', include_top=False)
install.run(self)
setup(name='dsbox-featurizer',
version='1.0.2',
url='https://github.com/usc-isi-i2/dsbox-featurizer',
maintainer_email='Ke-Thia Yao',
maintainer='kyao@isi.edu',
description='DSBox Featurization primitives',
license='MIT',
packages=[
'dsbox',
'dsbox.datapreprocessing',
'dsbox.datapreprocessing.featurizer',
'dsbox.datapreprocessing.featurizer.multiTable',
'dsbox.datapreprocessing.featurizer.image',
'dsbox.datapreprocessing.featurizer.pass',
'dsbox.datapreprocessing.featurizer.timeseries'],
zip_safe=False,
python_requires='>=3.6',
install_requires=[
'scipy>=0.19.0,<1.2', 'numpy>=1.11.1', 'pandas>=0.20.1',
'python-dateutil>=2.5.2', 'six>=1.10.0', 'stopit==1.1.2',
'scikit-learn>=0.18.0',
'Keras==2.2.4', 'Pillow==5.1.0', 'tensorflow==1.12', 'h5py<=2.7.1', "pyramid-arima==0.8.1"
],
# dependency_links=[
# 'git+https://github.com/usc-isi-i2/dsbox-cleaning.git@7106cf48d56eaa45460792edf7416dfd57548370#egg=dsbox-datacleaning-1.3.1'
# ],
keywords='d3m_primitive',
entry_points={
'd3m.primitives': [
'data_preprocessing.DoNothing.DSBOX = dsbox.datapreprocessing.featurizer.pass:DoNothing',
'data_preprocessing.DoNothingForDataset.DSBOX = dsbox.datapreprocessing.featurizer.pass:DoNothingForDataset',
'feature_extraction.MultiTableFeaturization.DSBOX = dsbox.datapreprocessing.featurizer.multiTable:MultiTableFeaturization',
'data_preprocessing.DataFrameToTensor.DSBOX = dsbox.datapreprocessing.featurizer.image:DataFrameToTensor',
'feature_extraction.Vgg16ImageFeature.DSBOX = dsbox.datapreprocessing.featurizer.image:Vgg16ImageFeature',
'feature_extraction.ResNet50ImageFeature.DSBOX = dsbox.datapreprocessing.featurizer.image:ResNet50ImageFeature',
'data_preprocessing.TimeseriesToList.DSBOX = dsbox.datapreprocessing.featurizer.timeseries:TimeseriesToList',
'feature_extraction.RandomProjectionTimeSeriesFeaturization.DSBOX = dsbox.datapreprocessing.featurizer.timeseries:RandomProjectionTimeSeriesFeaturization',
'data_transformation.GroupUpByTimeSeries.DSBOX = dsbox.datapreprocessing.featurizer.timeseries:GroupUpByTimeSeries',
'time_series_forecasting.Arima.DSBOX = dsbox.datapreprocessing.featurizer.timeseries:AutoArima',
'time_series_forecasting.RNNTimeSeries.DSBOX = dsbox.datapreprocessing.featurizer.timeseries:RNNTimeSeries'
],
},
cmdclass={
'develop': PostDevelopCommand,
'install': PostInstallCommand,
}
)