Method predict

crates/openquant/src/sb_bagging.rs:233–250

(&self, x: &DMatrix<f64>)

    }

    pub fn predict(&self, x: &DMatrix<f64>) -> Result<Vec<u8>, SbBaggingError> {
        if self.estimators.is_empty() {
            return Err(SbBaggingError::EmptyInput);
        }
        let mut out = vec![0u8; x.nrows()];
        for r in 0..x.nrows() {
            let mut votes = 0usize;
            for est in &self.estimators {
                let ge = x[(r, est.feature_idx)] >= est.threshold;
                let pred_one = if est.positive_on_ge { ge } else { !ge };
                if pred_one {
                    votes += 1;
                }
            }
            out[r] = if votes * 2 >= self.estimators.len() { 1 } else { 0 };
        }
        Ok(out)
    }
}

#[derive(Debug, Clone)]