Skip to main content

Ensemble

Trait Ensemble 

Source
pub trait Ensemble {
    // Required methods
    fn predict(&self, input: &Array2<f64>) -> TrainResult<Array2<f64>>;
    fn num_models(&self) -> usize;
}
Expand description

Trait for ensemble methods.

Required Methods§

Source

fn predict(&self, input: &Array2<f64>) -> TrainResult<Array2<f64>>

Predict using the ensemble.

§Arguments
  • input - Input data [batch_size, features]
§Returns

Ensemble predictions [batch_size, num_classes]

Source

fn num_models(&self) -> usize

Get the number of models in the ensemble.

Implementors§