hyperopt
/
setup.py
61 строка · 2.1 Кб
1import re
2
3import setuptools
4
5with open("hyperopt/__init__.py", encoding="utf8") as f:
6version = re.search(r"__version__ = \"(.*?)\"", f.read()).group(1)
7if version is None:
8raise ImportError("Could not find __version__ in hyperopt/__init__.py")
9
10setuptools.setup(
11name="hyperopt",
12version=version,
13packages=setuptools.find_packages(include=["hyperopt*"]),
14entry_points={"console_scripts": ["hyperopt-mongo-worker=hyperopt.mongoexp:main"]},
15url="https://hyperopt.github.io/hyperopt",
16project_urls={
17"Source": "https://github.com/hyperopt/hyperopt",
18},
19author="James Bergstra",
20author_email="james.bergstra@gmail.com",
21description="Distributed Asynchronous Hyperparameter Optimization",
22long_description="",
23classifiers=[
24"Development Status :: 3 - Alpha",
25"Intended Audience :: Education",
26"Intended Audience :: Science/Research",
27"Intended Audience :: Developers",
28"Environment :: Console",
29"License :: OSI Approved :: BSD License",
30"Operating System :: MacOS :: MacOS X",
31"Operating System :: Microsoft :: Windows",
32"Operating System :: POSIX",
33"Operating System :: Unix",
34"Programming Language :: Python",
35"Programming Language :: Python :: 3",
36"Programming Language :: Python :: 3 :: Only",
37"Programming Language :: Python :: 3.7",
38"Topic :: Scientific/Engineering",
39"Topic :: Software Development",
40],
41platforms=["Linux", "OS-X", "Windows"],
42license="BSD",
43keywords="Bayesian optimization hyperparameter model selection",
44include_package_data=True,
45requires_python=">=3.7",
46install_requires=[
47"numpy>=1.17",
48"scipy",
49"networkx>=2.2",
50"tqdm",
51"cloudpickle",
52],
53extras_require={
54"SparkTrials": ["pyspark", "py4j"],
55"MongoTrials": "pymongo>=4.0.0",
56"ATPE": ["lightgbm", "scikit-learn"],
57"dev": ["black", "pre-commit", "nose", "pytest"],
58},
59tests_require=["nose", "pytest"],
60zip_safe=False,
61)
62