Struct ScikitLearnModel 

pub struct ScikitLearnModel { /* private fields */ }

Generic scikit-learn compatible model wrapper

Implementations

impl ScikitLearnModel

pub fn linear_regression() -> ScikitLearnModel

Create a linear regression model with scikit-learn API

pub fn random_forest_classifier() -> ScikitLearnModel

Create a random forest classifier with scikit-learn API

pub fn svm_classifier() -> ScikitLearnModel

Create a support vector machine with scikit-learn API

Trait Implementations

impl Clone for ScikitLearnModel

fn clone(&self) -> ScikitLearnModel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ScikitLearnModel

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Estimator for ScikitLearnModel

type Config = HashMap<String, ParamValue>

Configuration type for the estimator

type Error = SklearsError

Error type for the estimator

type Float = f64

The numeric type used by this estimator

fn config(&self) -> &<ScikitLearnModel as Estimator>::Config

Get estimator configuration

fn validate_config(&self) -> Result<(), SklearsError>

Validate estimator configuration with detailed error context

fn check_compatibility( &self, n_samples: usize, n_features: usize, ) -> Result<(), SklearsError>

Check if estimator is compatible with given data dimensions

fn metadata(&self) -> EstimatorMetadata

Get estimator metadata

impl<'a> Fit<ArrayBase<ViewRepr<&'a f64>, Dim<[usize; 2]>>, ArrayBase<ViewRepr<&'a f64>, Dim<[usize; 1]>>> for ScikitLearnModel

type Fitted = FittedScikitLearnModel

The fitted model type

fn fit( self, x: &ArrayBase<ViewRepr<&'a f64>, Dim<[usize; 2]>>, y: &ArrayBase<ViewRepr<&'a f64>, Dim<[usize; 1]>>, ) -> Result<<ScikitLearnModel as Fit<ArrayBase<ViewRepr<&'a f64>, Dim<[usize; 2]>>, ArrayBase<ViewRepr<&'a f64>, Dim<[usize; 1]>>>>::Fitted, SklearsError>

Fit the model to the provided data with validation

fn fit_with_validation( self, x: &X, y: &Y, _x_val: Option<&X>, _y_val: Option<&Y>, ) -> Result<(Self::Fitted, FitMetrics), SklearsError>

where Self: Sized,

Fit with custom validation and early stopping

impl SklearnCompatible for ScikitLearnModel

fn set_param( &mut self, param: &str, value: impl Into<ParamValue>, ) -> Result<(), SklearsError>

Set a hyperparameter using string key-value pairs (scikit-learn style)

fn get_param(&self, param: &str) -> Result<ParamValue, SklearsError>

Get a hyperparameter value

fn get_params(&self, deep: bool) -> HashMap<String, ParamValue>

Get all hyperparameters as a dictionary

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

Set multiple parameters from a dictionary