hyperopt-gbt / hyperopt_gbt /__init__.py
erinkhoo's picture
Upload hyperopt_gbt/__init__.py
47fda1e verified
raw
history blame contribute delete
677 Bytes
"""
HyperOptimized Gradient Boosted Trees (HyperOpt-GBT)
A scikit-learn compatible gradient boosted tree library that synthesizes
best innovations from YDF, CatBoost, XGBoost, and LightGBM.
"""
__version__ = "0.1.0"
from .core import (
HyperOptGradientBoostedClassifier,
HyperOptGradientBoostedRegressor,
)
from .inference import (
compile_inference_engine,
NaiveEngine,
FlatTreeEngine,
BatchedSIMDEngine,
QuickScorerEngine,
)
__all__ = [
"HyperOptGradientBoostedClassifier",
"HyperOptGradientBoostedRegressor",
"compile_inference_engine",
"NaiveEngine",
"FlatTreeEngine",
"BatchedSIMDEngine",
"QuickScorerEngine",
]