hyperopt_gbt/__init__.py

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