Struct rusty_ai::forests::regressor::RandomForestRegressor

pub struct RandomForestRegressor<T: RealNumber> { /* private fields */ }

Implementations

impl<T: RealNumber> RandomForestRegressor<T>

pub fn new() -> Self

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

pub fn set_trees(&mut self, trees: Vec<DecisionTreeRegressor<T>>)

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 trees(&self) -> &Vec<DecisionTreeRegressor<T>>

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 fit( &mut self, dataset: &Dataset<T, T>, seed: Option<u64> ) -> Result<String, Box<dyn Error>>

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

Trait Implementations

impl<T: Clone + RealNumber> Clone for RandomForestRegressor<T>

fn clone(&self) -> RandomForestRegressor<T>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<T: Debug + RealNumber> Debug for RandomForestRegressor<T>

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

Formats the value using the given formatter.

impl<T: RealNumber> Default for RandomForestRegressor<T>

fn default() -> Self

Returns the “default value” for a type.

impl<T: RealNumber> RegressionMetrics<T> for RandomForestRegressor<T>

fn mse( &self, y_true: &DVector<T>, y_pred: &DVector<T> ) -> Result<T, Box<dyn Error>>

fn mae( &self, y_true: &DVector<T>, y_pred: &DVector<T> ) -> Result<T, Box<dyn Error>>

fn r2( &self, y_true: &DVector<T>, y_pred: &DVector<T> ) -> Result<T, Box<dyn Error>>