Struct rusty_ai::trees::classifier::DecisionTreeClassifier

pub struct DecisionTreeClassifier<XT: Number, YT: WholeNumber> { /* private fields */ }

Decision Tree Classifier

Implementations

impl<XT: Number, YT: WholeNumber> DecisionTreeClassifier<XT, YT>

pub fn new() -> Self

pub fn with_params( criterion: Option<String>, min_samples_split: Option<u16>, max_depth: Option<u16> ) -> Result<Self, Box<dyn Error>>

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

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

pub fn set_criterion(&mut self, criterion: String) -> Result<(), Box<dyn Error>>

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

pub fn min_samples_split(&self) -> u16

pub fn criterion(&self) -> &str

pub fn fit( &mut self, dataset: &Dataset<XT, YT> ) -> Result<String, Box<dyn Error>>

Build the tree from a dataset.

• dataset - dataset containing features and labels

pub fn predict( &self, features: &DMatrix<XT> ) -> Result<DVector<YT>, Box<dyn Error>>

Predict the labels for new data.

• features - MxN matrix for M

Trait Implementations

impl<XT: Clone + Number, YT: Clone + WholeNumber> Clone for DecisionTreeClassifier<XT, YT>

fn clone(&self) -> DecisionTreeClassifier<XT, YT>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<XT: Debug + Number, YT: Debug + WholeNumber> Debug for DecisionTreeClassifier<XT, YT>

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

Formats the value using the given formatter.

impl<XT: Number, YT: WholeNumber> Default for DecisionTreeClassifier<XT, YT>

fn default() -> Self

Returns the “default value” for a type.