TrainedBaseModel

Trait TrainedBaseModel 

Source
pub trait TrainedBaseModel:
    Send
    + Sync
    + Debug {
    // Required method
    fn predict(
        &self,
        x: &ArrayBase<ViewRepr<&f64>, Dim<[usize; 2]>>,
    ) -> Result<ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>, SklearsError>;

    // Provided method
    fn get_importance(
        &self,
    ) -> Option<ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>>> { ... }
}
Expand description

Trait for trained base models

Required Methods§

Provided Methods§

Implementors§