| 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"], |
| }, |
| ) |
|
|
