Struct rusty_ai::trees::params::TreeParams

pub struct TreeParams {
    pub min_samples_split: u16,
    pub max_depth: Option<u16>,
}

Struct representing the parameters for a decision tree.

Fields

min_samples_split: u16

max_depth: Option<u16>

Implementations

impl TreeParams

pub fn new() -> Self

Creates a new instance of TreeParams with default values.

pub fn set_min_samples_split( &mut self, min_samples_split: u16 ) -> Result<(), Box<dyn Error>>

Sets the minimum number of samples required to split a node.

Arguments

• min_samples_split - The minimum number of samples to split.

Errors

Returns an error if min_samples_split is less than 2.

pub fn set_max_depth( &mut self, max_depth: Option<u16> ) -> Result<(), Box<dyn Error>>

Sets the maximum depth of the decision tree.

Arguments

• max_depth - The maximum depth of the tree.

Errors

Returns an error if max_depth is less than 1.

pub fn min_samples_split(&self) -> u16

Returns the minimum number of samples required to split a node.

pub fn max_depth(&self) -> Option<u16>

Returns the maximum depth of the decision tree.

Trait Implementations

impl Clone for TreeParams

fn clone(&self) -> TreeParams

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for TreeParams

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

Formats the value using the given formatter.

impl Default for TreeParams

fn default() -> Self

Creates a new instance of TreeParams with default values.