Trait ClassicalModel 

pub trait ClassicalModel: Send + Sync {
    // Required methods
    fn fit(&mut self, X: &ArrayD<f64>, y: &ArrayD<f64>) -> Result<()>;
    fn predict(&self, X: &ArrayD<f64>) -> Result<ArrayD<f64>>;
    fn get_params(&self) -> HashMap<String, f64>;
    fn set_params(&mut self, params: HashMap<String, f64>) -> Result<()>;
    fn feature_importance(&self) -> Option<Array1<f64>>;
}

Classical model trait for integration

Required Methods

fn fit(&mut self, X: &ArrayD<f64>, y: &ArrayD<f64>) -> Result<()>

Train the model

fn predict(&self, X: &ArrayD<f64>) -> Result<ArrayD<f64>>

Make predictions

fn get_params(&self) -> HashMap<String, f64>

Get model parameters

fn set_params(&mut self, params: HashMap<String, f64>) -> Result<()>

Set model parameters

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

Get feature importance (if available)

Implementors