Struct DecisionTreeClassifier 

pub struct DecisionTreeClassifier { /* private fields */ }

Decision tree classifier using the CART algorithm.

Uses Gini impurity for splitting criterion and builds trees recursively.

Implementations

impl DecisionTreeClassifier

pub fn new() -> Self

Creates a new decision tree classifier with default parameters.

pub fn with_max_depth(self, depth: usize) -> Self

Sets the maximum depth of the tree.

Arguments

• depth - Maximum depth (root has depth 0)

pub fn fit(&mut self, x: &Matrix<f32>, y: &[usize]) -> Result<()>

Fits the decision tree to training data.

Arguments

• x - Training features (n_samples × n_features)
• y - Training labels (n_samples class indices)

Errors

Returns an error if the data is invalid.

pub fn predict(&self, x: &Matrix<f32>) -> Vec<usize>

Predicts class labels for samples.

Arguments

• x - Feature matrix (n_samples × n_features)

Returns

Vector of predicted class labels

Panics

Panics if called before fit() or if feature count doesn’t match training data

pub fn score(&self, x: &Matrix<f32>, y: &[usize]) -> f32

Computes the accuracy score on test data.

Arguments

• x - Test features (n_samples × n_features)
• y - True labels (n_samples)

Returns

Accuracy (fraction of correct predictions)

pub fn save<P: AsRef<Path>>(&self, path: P) -> Result<(), String>

Saves the model to a binary file using bincode.

Errors

Returns an error if serialization or file writing fails.

pub fn load<P: AsRef<Path>>(path: P) -> Result<Self, String>

Loads a model from a binary file.

Errors

Returns an error if file reading or deserialization fails.

pub fn save_safetensors<P: AsRef<Path>>(&self, path: P) -> Result<(), String>

Saves the model to SafeTensors format.

Serializes the tree structure as flat arrays using pre-order traversal.

Errors

Returns an error if:

• Model is not fitted
• Serialization fails
• File writing fails

pub fn load_safetensors<P: AsRef<Path>>(path: P) -> Result<Self, String>

Loads a model from SafeTensors format.

Errors

Returns an error if:

• File reading fails
• SafeTensors format is invalid
• Required tensors are missing

Trait Implementations

impl Clone for DecisionTreeClassifier

fn clone(&self) -> DecisionTreeClassifier

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DecisionTreeClassifier

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

Formats the value using the given formatter.

impl Default for DecisionTreeClassifier

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for DecisionTreeClassifier

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for DecisionTreeClassifier

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.