Struct stamm::randforest::RandomForest

pub struct RandomForest<L, F> where
    F: TreeFunction,  { /* fields omitted */ }

A random forest to combine some decision trees. The decision trees have leafs with data from type L and use a TreeFunction from type F.

Methods

impl<L, F> RandomForest<L, F> where
    F: TreeFunction, 

pub fn forest_predictions(&self, input: &F::Data) -> Vec<&L>

Let every tree predict a result of the input. Returns the results in form a Vec of the leaf data.

pub fn predict<V>(&self, input: &F::Data, voting_method: V) -> Option<f64> where
    V: VotingMethod<L>, 

Let every tree predict a result and combine them using a vote method.

impl<L, F> RandomForest<L, F> where
    F: TreeFunction + Send + Sync,
    <F as TreeFunction>::Param: Send + Sync,
    <F as TreeFunction>::Data: Send + Sync,
    L: Send + Sync, 

pub fn forest_predictions_parallel(&self, input: &F::Data) -> Vec<&L>

Like forest_predictions but use rayon to parallelize the computation.

Trait Implementations