Struct GradientBoostingClassifier 

pub struct GradientBoostingClassifier<F> {
    pub n_estimators: usize,
    pub learning_rate: f64,
    pub max_depth: Option<usize>,
    pub min_samples_split: usize,
    pub min_samples_leaf: usize,
    pub subsample: f64,
    pub loss: ClassificationLoss,
    pub random_state: Option<u64>,
    /* private fields */
}

Gradient boosting classifier.

For binary classification a single model is trained on log-odds residuals. For multiclass (K classes), K regression trees are built per boosting round (one-vs-rest in probability space via softmax).

Type Parameters

• F: The floating-point type (f32 or f64).

Fields

n_estimators: usize

Number of boosting stages.

learning_rate: f64

Learning rate (shrinkage).

max_depth: Option<usize>

Maximum depth of each tree.

min_samples_split: usize

Minimum number of samples required to split an internal node.

min_samples_leaf: usize

Minimum number of samples required in a leaf node.

subsample: f64

Fraction of samples to use for fitting each tree.

loss: ClassificationLoss

Classification loss function.

random_state: Option<u64>

Random seed for reproducibility.

Implementations

impl<F: Float> GradientBoostingClassifier<F>

pub fn new() -> Self

Create a new GradientBoostingClassifier with default settings.

Defaults: n_estimators = 100, learning_rate = 0.1, max_depth = Some(3), min_samples_split = 2, min_samples_leaf = 1, subsample = 1.0, loss = LogLoss.

pub fn with_n_estimators(self, n: usize) -> Self

Set the number of boosting stages.

pub fn with_learning_rate(self, lr: f64) -> Self

Set the learning rate (shrinkage).

pub fn with_max_depth(self, d: Option<usize>) -> Self

Set the maximum tree depth.

pub fn with_min_samples_split(self, n: usize) -> Self

Set the minimum number of samples to split a node.

pub fn with_min_samples_leaf(self, n: usize) -> Self

Set the minimum number of samples in a leaf.

pub fn with_subsample(self, ratio: f64) -> Self

Set the subsample ratio.

pub fn with_random_state(self, seed: u64) -> Self

Set the random seed for reproducibility.

Trait Implementations

impl<F: Clone> Clone for GradientBoostingClassifier<F>

fn clone(&self) -> GradientBoostingClassifier<F>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<F: Debug> Debug for GradientBoostingClassifier<F>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<F: Float> Default for GradientBoostingClassifier<F>

fn default() -> Self

Returns the “default value” for a type.

impl<F: Float + Send + Sync + 'static> Fit<ArrayBase<OwnedRepr<F>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<usize>, Dim<[usize; 1]>>> for GradientBoostingClassifier<F>

fn fit( &self, x: &Array2<F>, y: &Array1<usize>, ) -> Result<FittedGradientBoostingClassifier<F>, FerroError>

Fit the gradient boosting classifier.

Errors

Returns FerroError::ShapeMismatch if x and y have different numbers of samples. Returns FerroError::InsufficientSamples if there are no samples. Returns FerroError::InvalidParameter for invalid hyperparameters.

type Fitted = FittedGradientBoostingClassifier<F>

The fitted model type returned by fit.

type Error = FerroError

The error type returned by fit.

impl<F: Float + ToPrimitive + FromPrimitive + Send + Sync + 'static> PipelineEstimator<F> for GradientBoostingClassifier<F>

fn fit_pipeline( &self, x: &Array2<F>, y: &Array1<F>, ) -> Result<Box<dyn FittedPipelineEstimator<F>>, FerroError>

Fit this estimator on the given data.