Trait Classifier

pub trait Classifier<Features, Label>
where
    Label: Clone,
{
    // Required methods
    fn labels(&self) -> &[Label];
    fn predict_proba(&self, arr: &Features) -> Option<Array2<f64>>;

    // Provided method
    fn predict(&self, arr: &Features) -> Option<Vec<Label>> { ... }
}

Trait to interface with a fitted classification model

Required Methods

fn labels(&self) -> &[Label]

Labels on which the model is fitted.

fn predict_proba(&self, arr: &Features) -> Option<Array2<f64>>

Estimates likelihood of each class per record. Rows correspond to each record, columns are in the same order as label function.

Provided Methods

fn predict(&self, arr: &Features) -> Option<Vec<Label>>

Provided function which returns the most likely class per record based on the results of predict_proba().

Implementors

impl<Label: Clone> Classifier<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>, Label> for GaussianNB<Label>