File size: 1,137 Bytes
bb6bd22 | 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 | from setuptools import setup, find_packages
setup(
name="hyperopt-gbt",
version="0.1.0",
author="HyperOpt-GBT Team",
description="HyperOptimized Gradient Boosted Trees - combining best innovations from YDF, CatBoost, XGBoost, and LightGBM",
url="https://huggingface.co/erinkhoo/hyperopt-gbt",
packages=find_packages(),
classifiers=[
"Development Status :: 3 - Alpha",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: Apache Software License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
python_requires=">=3.8",
install_requires=[
"numpy>=1.20.0",
"numba>=0.56.0",
"scikit-learn>=1.0.0",
"pandas>=1.3.0",
],
extras_require={
"benchmark": ["xgboost", "lightgbm", "catboost"],
"dev": ["pytest", "black", "flake8"],
},
)
|