erinkhoo commited on
Commit
bb6bd22
·
verified ·
1 Parent(s): b5f7206

Upload setup.py

Browse files
Files changed (1) hide show
  1. setup.py +32 -0
setup.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from setuptools import setup, find_packages
2
+
3
+ setup(
4
+ name="hyperopt-gbt",
5
+ version="0.1.0",
6
+ author="HyperOpt-GBT Team",
7
+ description="HyperOptimized Gradient Boosted Trees - combining best innovations from YDF, CatBoost, XGBoost, and LightGBM",
8
+ url="https://huggingface.co/erinkhoo/hyperopt-gbt",
9
+ packages=find_packages(),
10
+ classifiers=[
11
+ "Development Status :: 3 - Alpha",
12
+ "Intended Audience :: Science/Research",
13
+ "License :: OSI Approved :: Apache Software License",
14
+ "Programming Language :: Python :: 3",
15
+ "Programming Language :: Python :: 3.8",
16
+ "Programming Language :: Python :: 3.9",
17
+ "Programming Language :: Python :: 3.10",
18
+ "Programming Language :: Python :: 3.11",
19
+ "Topic :: Scientific/Engineering :: Artificial Intelligence",
20
+ ],
21
+ python_requires=">=3.8",
22
+ install_requires=[
23
+ "numpy>=1.20.0",
24
+ "numba>=0.56.0",
25
+ "scikit-learn>=1.0.0",
26
+ "pandas>=1.3.0",
27
+ ],
28
+ extras_require={
29
+ "benchmark": ["xgboost", "lightgbm", "catboost"],
30
+ "dev": ["pytest", "black", "flake8"],
31
+ },
32
+ )