-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
34 lines (33 loc) · 1.04 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
from setuptools import setup, find_packages
setup(
name='predictor',
version='0.1.0',
packages=find_packages(),
entry_points={
'console_scripts': [
'predictor=app.main:main'
],
'predictor.plugins': [
'default=app.plugins.predictor_plugin_ann:Plugin',
'ann=app.plugins.predictor_plugin_ann:Plugin',
'cnn=app.plugins.predictor_plugin_cnn:Plugin',
'lstm=app.plugins.predictor_plugin_lstm:Plugin',
'transformer=app.plugins.predictor_plugin_transformer:Plugin'
]
},
install_requires=[
'tensorflow-gpu',
'nvidia-cudnn-cu11',
'numpy',
'pandas',
'h5py',
'scipy',
'build',
'pytest',
'pdoc3',
'keras-multi-head'
],
author='Harvey Bastidas',
author_email='[email protected]',
description='A timeseries prediction system that supports dynamic loading of predictor plugins for processing time series data.'
)