Skip to main content

HistGradientBoostingRegressor

Histogram-based gradient boosting for regression — the fast, large-data GB path.

Algorithm

Features are quantile-binned, then squared-error boosting proceeds on the binned representation with histogram split finding.

Constructor

Skigen::HistGradientBoostingRegressor<Scalar> model(Scalar learning_rate = 0.1, int max_iter = 100, ...);

Parameters

ParameterDefaultDescription
learning_rate0.1Shrinkage per iteration.
max_iter100Number of iterations.
max_bins255Bin resolution.
max_leaf_nodes31Leaves per tree.
random_statenulloptSeed.

Methods

MethodDescription
fit(X, y)Bin features, then boost.
predict(X)Boosted prediction.
score(X, y)R².

Fitted Attributes

AccessorDescription
bin_edges()Per-feature quantile bin edges.
train_score()Per-iteration training MSE.

Example

Skigen::HistGradientBoostingRegressor<double> gb;
gb.fit(X, y);
auto preds = gb.predict(X_test);
Verified against scikit-learn

This estimator is checked by the parity suite. See the generator tests/parity/generate_ensemble_reference.py and the reference fixtures in tests/parity/data/hist_gradient_boosting_regressor/, exercised by tests/parity/parity_ensemble.cpp.

API Reference

For full signatures see the HistGradientBoostingRegressor API Reference.