Struct rusty_ai::forests::classifier::RandomForestClassifier

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

Implementations

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

pub fn new() -> Self

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

pub fn set_trees(&mut self, trees: Vec<DecisionTreeClassifier<XT, YT>>)

pub fn set_num_trees(&mut self, num_trees: usize) -> Result<(), Box<dyn Error>>

pub fn set_sample_size( &mut self, sample_size: Option<usize> ) -> Result<(), 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 trees(&self) -> &Vec<DecisionTreeClassifier<XT, YT>>

pub fn num_trees(&self) -> usize

pub fn sample_size(&self) -> Option<usize>

pub fn min_samples_split(&self) -> u16

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

pub fn criterion(&self) -> &String

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

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

Trait Implementations

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

fn clone(&self) -> RandomForestClassifier<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 RandomForestClassifier<XT, YT>

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

Formats the value using the given formatter.

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

fn default() -> Self

Returns the “default value” for a type.