Struct FittedGradientBoostingRegressor 

pub struct FittedGradientBoostingRegressor<F> { /* private fields */ }

A fitted gradient boosting regressor.

Stores the initial prediction (intercept) and the sequence of fitted trees. Predictions are computed as init + learning_rate * sum(tree_predictions).

Implementations

impl<F: Float + Send + Sync + 'static> FittedGradientBoostingRegressor<F>

pub fn init(&self) -> F

Returns the initial prediction (intercept) of the boosted model.

pub fn learning_rate(&self) -> F

Returns the learning rate used during training.

pub fn trees(&self) -> &[Vec<Node<F>>]

Returns a reference to the sequence of fitted trees.

pub fn n_features(&self) -> usize

Returns the number of features the model was trained on.

Trait Implementations

impl<F: Clone> Clone for FittedGradientBoostingRegressor<F>

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

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<F: Debug> Debug for FittedGradientBoostingRegressor<F>

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

Formats the value using the given formatter.

impl<F: Float + Send + Sync + 'static> FittedPipelineEstimator<F> for FittedGradientBoostingRegressor<F>

fn predict_pipeline(&self, x: &Array2<F>) -> Result<Array1<F>, FerroError>

Generate predictions for the input data.

impl<F: Float + Send + Sync + 'static> HasFeatureImportances<F> for FittedGradientBoostingRegressor<F>

fn feature_importances(&self) -> &Array1<F>

Returns a reference to the feature importance array.

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

fn predict(&self, x: &Array2<F>) -> Result<Array1<F>, FerroError>

Predict target values.

Errors

Returns FerroError::ShapeMismatch if the number of features does not match the fitted model.

type Output = ArrayBase<OwnedRepr<F>, Dim<[usize; 1]>>

The prediction output type (e.g., ndarray::Array1<F> or ndarray::Array1<usize>).

type Error = FerroError

The error type returned by predict.