Struct DecisionTreeBuilder 

pub struct DecisionTreeBuilder<State> { /* private fields */ }

Builder pattern for configuring DecisionTree

Implementations

impl<State> DecisionTreeBuilder<State>

pub fn new() -> Self

Create a new builder

pub fn criterion(self, criterion: SplitCriterion) -> Self

Set the split criterion

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

Set the maximum depth of the tree

pub fn min_samples_split(self, min_samples_split: usize) -> Self

Set the minimum samples required to split an internal node

pub fn min_samples_leaf(self, min_samples_leaf: usize) -> Self

Set the minimum samples required to be at a leaf node

pub fn min_impurity_decrease(self, min_impurity_decrease: f64) -> Self

Set the minimum impurity decrease required for a split

pub fn max_features(self, _max_features: Option<usize>) -> Self

Set the maximum number of features to consider for splitting

pub fn random_state(self, random_state: Option<u64>) -> Self

Set the random seed

pub fn build(self) -> DecisionTree<State>

Build the DecisionTree with the configured parameters

Trait Implementations

impl<State: Debug> Debug for DecisionTreeBuilder<State>

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

Formats the value using the given formatter.

impl<State> Default for DecisionTreeBuilder<State>

fn default() -> Self

Returns the “default value” for a type.