-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
47 lines (46 loc) · 1.41 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
from setuptools import setup
setup(
name = 'nestedhyperboost',
version = '0.0.2',
description = 'A wrapper for conducting Nested Cross-Validation with Bayesian Hyper-Parameter Optimized Gradient Boosting',
long_description = open('README.md').read(),
long_description_content_type = "text/markdown",
author = 'Nick Kunz',
author_email = 'nick.kunz@me.com',
url = 'https://github.com/nickkunz/nestedhyperboost',
classifiers = [
'Programming Language :: Python :: 3',
'Intended Audience :: Science/Research',
'Topic :: Scientific/Engineering',
'Topic :: Scientific/Engineering :: Mathematics',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'License :: OSI Approved :: GNU General Public License v3 (GPLv3)'
],
keywords = [
'nested cross-validation',
'bayesian optimization',
'gradient boosting',
'xgboost',
'lightgbm',
'catboost'
],
packages = [
'nestedhyperboost',
'nestedhyperboost.xgboost',
'nestedhyperboost.lightgbm',
'nestedhyperboost.catboost'
],
include_package_data = True,
install_requires = [
'numpy',
'pandas',
'matplotlib',
'sklearn',
'hyperopt',
'xgboost',
'lightgbm',
'catboost'
],
tests_require = ['nose'],
test_suite = 'nose.collector'
)