Struct RandomForestClassifier 

pub struct RandomForestClassifier {
    pub n_estimators: usize,
    pub max_depth: Option<usize>,
    pub min_samples_split: usize,
    pub min_samples_leaf: usize,
    pub max_features: Option<usize>,
    pub bootstrap: bool,
    pub n_jobs: Option<usize>,
    /* private fields */
}

Random Forest Classifier

Fields

n_estimators: usize

max_depth: Option<usize>

min_samples_split: usize

min_samples_leaf: usize

max_features: Option<usize>

bootstrap: bool

n_jobs: Option<usize>

Implementations

impl RandomForestClassifier

pub fn new(n_estimators: usize) -> Self

pub fn max_depth(self, depth: usize) -> Self

pub fn min_samples_split(self, n: usize) -> Self

pub fn max_features(self, n: usize) -> Self

pub fn fit(&mut self, x: &Tensor, y: &Tensor)

pub fn predict(&self, x: &Tensor) -> Tensor

pub fn predict_proba(&self, x: &Tensor) -> Tensor

pub fn feature_importances(&self, n_features: usize) -> Vec<f32>

Feature importance based on impurity decrease