Enum DecisionTree 

pub enum DecisionTree {
    Leaf(usize),
    Split {
        stump: DecisionStump,
        left: Box<DecisionTree>,
        right: Box<DecisionTree>,
    },
}

A binary decision tree of bounded depth.

Variants

Leaf(usize)

Leaf node with a class label.

Split

Internal split node.

Fields

stump: DecisionStump

The decision stump at this node.

left: Box<DecisionTree>

Left subtree (feature < threshold).

right: Box<DecisionTree>

Right subtree (feature >= threshold).

Implementations

impl DecisionTree

pub fn train(features: &[Vec<f64>], labels: &[usize], max_depth: usize) -> Self

Build a decision tree from labelled data with a maximum depth.

pub fn predict(&self, features: &[f64]) -> usize

Predict the class for a single feature vector.

Trait Implementations

impl Clone for DecisionTree

fn clone(&self) -> DecisionTree

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DecisionTree

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

Formats the value using the given formatter.