Struct FittedDecisionTreeClassifier 

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

A fitted CART decision tree classifier.

Stores the learned tree as a flat Vec<Node<F>> for cache-friendly traversal. Implements Predict for generating class predictions and HasFeatureImportances for inspecting per-feature importance scores.

Implementations

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

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

Returns a reference to the flat node storage of the tree.

pub fn n_features(&self) -> usize

Returns the number of features the model was trained on.

pub fn predict_proba(&self, x: &Array2<F>) -> Result<Array2<F>, FerroError>

Predict class probabilities for each sample.

Returns a 2-D array of shape (n_samples, n_classes).

Errors

Returns FerroError::ShapeMismatch if the number of features does not match the training data.

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

Mean accuracy on the given test data and labels. Equivalent to sklearn’s ClassifierMixin.score.

Errors

Returns FerroError::ShapeMismatch if x.nrows() != y.len() or the feature count does not match the training data.

pub fn predict_log_proba(&self, x: &Array2<F>) -> Result<Array2<F>, FerroError>

Element-wise log of predict_proba. Mirrors sklearn’s ClassifierMixin.predict_log_proba.

Errors

Forwards any error from predict_proba.

Trait Implementations

impl<F: Clone> Clone for FittedDecisionTreeClassifier<F>

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

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<F: Debug> Debug for FittedDecisionTreeClassifier<F>

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

Formats the value using the given formatter.

impl<F: Float + Send + Sync + 'static> HasClasses for FittedDecisionTreeClassifier<F>

fn classes(&self) -> &[usize]

Returns the sorted list of unique class labels.

fn n_classes(&self) -> usize

Returns the number of distinct classes.

impl<F: Float + Send + Sync + 'static> HasFeatureImportances<F> for FittedDecisionTreeClassifier<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 FittedDecisionTreeClassifier<F>

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

Predict class labels for the given feature matrix.

Errors

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

type Output = ArrayBase<OwnedRepr<usize>, 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.