-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
45 lines (44 loc) · 933 Bytes
/
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
from setuptools import find_packages, setup
setup(
name='mergernet',
version='0.2.0',
description='deep learning model',
author='Natanael',
author_email='nauxmac@gmail.com',
packages=find_packages(),
include_package_data=True,
package_data={
'mergernet': []
},
install_requires=[
'wheel',
'numpy>=1.21.6',
'pandas>=1.3.5',
'tensorflow>=2.9.2',
'scikit-learn>=1.0.2',
'tqdm>=4.64.1',
'requests>=2.23',
'Pillow>=7.1.2',
'astropy>=5.2',
'tensorflow_addons',
'tensorboard',
'optuna',
'plotly',
'wandb',
'zoobot==0.0.4',
'keras_applications', # zoobot
'tensorflow_probability>=0.11', # zoobot
'seaborn' , # zoobot
'pydot>=1.4.2', # tf.keras.utils.plot_model
],
extras_require={
'docs': [
'Jinja2>=3.1',
'sphinxcontrib-napoleon',
'sphinx',
'furo',
'ipykernel',
'sphinx_copybutton',
]
}
)