Struct AdaBoostClassifier 

pub struct AdaBoostClassifier {
    pub n_estimators: usize,
    pub learning_rate: f32,
    pub algorithm: AdaBoostAlgorithm,
    /* private fields */
}

AdaBoost Classifier (Adaptive Boosting)

Fields

n_estimators: usize

learning_rate: f32

algorithm: AdaBoostAlgorithm

Implementations

impl AdaBoostClassifier

pub fn new(n_estimators: usize) -> Self

pub fn learning_rate(self, lr: f32) -> Self

pub fn algorithm(self, algo: AdaBoostAlgorithm) -> Self

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

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

pub fn score(&self, x: &Tensor, y: &Tensor) -> f32