Struct ModelSelector 

pub struct ModelSelector<T: FloatBounds + ScalarOperand + ToPrimitive + Sum> { /* private fields */ }

Model selection and comparison framework

Implementations

impl<T: FloatBounds + ScalarOperand + ToPrimitive + Sum> ModelSelector<T>

pub fn new(cv_config: CrossValidationConfig) -> Self

Create a new model selector

pub fn cross_validate_classifier( &self, X: &Array2<T>, y: &[usize], hidden_layer_sizes: &[usize], activation: Activation, learning_rate: T, alpha: T, max_iter: usize, ) -> Result<CrossValidationResults<T>, SklearsError>

Perform k-fold cross-validation on a classifier

pub fn cross_validate_regressor( &self, X: &Array2<T>, y: &Array1<T>, hidden_layer_sizes: &[usize], activation: Activation, learning_rate: T, alpha: T, max_iter: usize, ) -> Result<CrossValidationResults<T>, SklearsError>

Perform k-fold cross-validation on a regressor

pub fn grid_search_classifier( &self, X: &Array2<T>, y: &[usize], config: GridSearchConfig<T>, ) -> Result<GridSearchResults<T>, SklearsError>

Perform grid search for classifier

pub fn compare_models( &mut self, models: Vec<ModelMetrics<T>>, ) -> Vec<ModelMetrics<T>>

Compare multiple models

pub fn get_rankings(&self) -> Vec<(usize, &ModelMetrics<T>)>

Get model rankings

Trait Implementations

impl<T: Debug + FloatBounds + ScalarOperand + ToPrimitive + Sum> Debug for ModelSelector<T>

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

Formats the value using the given formatter.