Skip to main content

CalibrableClassifier

scry_learn::calibration

Trait CalibrableClassifier 

Source 
pub trait CalibrableClassifier {
    // Required methods
    fn fit(&mut self, data: &Dataset) -> Result<()>;
    fn predict(&self, features: &[Vec<f64>]) -> Result<Vec<f64>>;
    fn predict_proba(&self, features: &[Vec<f64>]) -> Result<Vec<Vec<f64>>>;
    fn clone_box(&self) -> Box<dyn CalibrableClassifier>;
}
Expand description

Trait for classifiers that can be calibrated.

Any classifier with fit, predict, and predict_proba methods that returns Vec<Vec<f64>> for probabilities.

Required Methods§

Source

fn fit(&mut self, data: &Dataset) -> Result<()>

Train on a dataset.

Source

fn predict(&self, features: &[Vec<f64>]) -> Result<Vec<f64>>

Predict class labels.

Source

fn predict_proba(&self, features: &[Vec<f64>]) -> Result<Vec<Vec<f64>>>

Predict class probabilities. Returns [n_samples][n_classes].

Source

fn clone_box(&self) -> Box<dyn CalibrableClassifier>

Clone into a boxed trait object.

Implementors§