Trait PluginEstimator 

pub trait PluginEstimator: PluginComponent {
    // Required methods
    fn fit(
        &mut self,
        x: &ArrayView2<'_, Float>,
        y: &ArrayView1<'_, Float>,
    ) -> SklResult<()>;
    fn predict(&self, x: &ArrayView2<'_, Float>) -> SklResult<Array1<f64>>;
    fn score(
        &self,
        x: &ArrayView2<'_, Float>,
        y: &ArrayView1<'_, Float>,
    ) -> SklResult<f64>;
    fn is_fitted(&self) -> bool;

    // Provided methods
    fn predict_proba(&self, x: &ArrayView2<'_, Float>) -> SklResult<Array2<f64>> { ... }
    fn feature_importances(&self) -> Option<Array1<f64>> { ... }
}

Plugin-based estimator

Required Methods

fn fit( &mut self, x: &ArrayView2<'_, Float>, y: &ArrayView1<'_, Float>, ) -> SklResult<()>

Fit the estimator

fn predict(&self, x: &ArrayView2<'_, Float>) -> SklResult<Array1<f64>>

Predict using the fitted estimator

fn score( &self, x: &ArrayView2<'_, Float>, y: &ArrayView1<'_, Float>, ) -> SklResult<f64>

Score the estimator

fn is_fitted(&self) -> bool

Check if estimator is fitted

Provided Methods

fn predict_proba(&self, x: &ArrayView2<'_, Float>) -> SklResult<Array2<f64>>

Predict probabilities (for classifiers)

fn feature_importances(&self) -> Option<Array1<f64>>

Get feature importance (if available)

Implementors