Interface SimpleClassifier

crates/openquant/src/cross_validation.rs:4–10

Simple classifier interface for cross-validation.

/// Simple classifier interface for cross-validation.
pub trait SimpleClassifier {
    fn fit(&mut self, x: &[Vec<f64>], y: &[f64], sample_weight: Option<&[f64]>);
    fn predict_proba(&self, x: &[Vec<f64>]) -> Vec<f64>;
    fn predict(&self, x: &[Vec<f64>]) -> Vec<f64> {
        self.predict_proba(x).into_iter().map(|p| if p >= 0.5 { 1.0 } else { 0.0 }).collect()
    }
}

#[derive(Clone, Copy)]
pub enum Scoring {