Struct VotingClassifier 

pub struct VotingClassifier { /* private fields */ }

Voting Classifier for ensemble learning

Implementations

impl VotingClassifier

pub fn new(classifiers: Vec<(String, Box<dyn SklearnClassifier>)>) -> Self

Create new voting classifier

pub fn voting(self, voting: &str) -> Self

Set voting mode

pub fn weights(self, weights: Vec<f64>) -> Self

Set weights

Trait Implementations

impl SklearnClassifier for VotingClassifier

fn predict(&self, X: &Array2<f64>) -> Result<Array1<i32>>

Predict class labels

fn predict_proba(&self, X: &Array2<f64>) -> Result<Array2<f64>>

Predict class probabilities

fn classes(&self) -> &[i32]

Get unique class labels

fn score(&self, X: &Array2<f64>, y: &Array1<i32>) -> Result<f64>

Score the model (accuracy by default)

fn feature_importances(&self) -> Option<Array1<f64>>

Get feature importances (optional)

fn save(&self, _path: &str) -> Result<()>

Save model to file (optional)

impl SklearnEstimator for VotingClassifier

fn fit(&mut self, _X: &Array2<f64>, _y: Option<&Array1<f64>>) -> Result<()>

Fit the model to training data

fn get_params(&self) -> HashMap<String, String>

Get model parameters

fn set_params(&mut self, params: HashMap<String, String>) -> Result<()>

Set model parameters

fn is_fitted(&self) -> bool

Check if model is fitted

fn get_feature_names_out(&self) -> Vec<String>

Get feature names