MetaModel

scirs2_series::ensemble_automl

Trait MetaModel 

Source 
pub trait MetaModel<F: Float + Debug>: Debug {
    // Required methods
    fn train(
        &mut self,
        predictions: &Array2<F>,
        targets: &Array1<F>,
    ) -> Result<()>;
    fn predict(&self, predictions: &Array2<F>) -> Result<Array1<F>>;
    fn confidence(&self) -> F;
}
Expand description

Trait for meta-models used in stacking

Required Methods§

Source

fn train(&mut self, predictions: &Array2<F>, targets: &Array1<F>) -> Result<()>

Train the meta-model on base model predictions

Source

fn predict(&self, predictions: &Array2<F>) -> Result<Array1<F>>

Predict using the meta-model

Source

fn confidence(&self) -> F

Get model confidence

Implementors§