Struct SelectFromModel 

pub struct SelectFromModel<E, State = Untrained> { /* private fields */ }

Select features from any estimator with feature importances

Implementations

impl<E: Clone> SelectFromModel<E, Untrained>

pub fn new(estimator: E) -> Self

Create a new SelectFromModel selector

pub fn threshold(self, threshold: f64) -> Self

Set the threshold for feature selection If None, will use median of feature importances

pub fn max_features(self, max_features: usize) -> Self

Set the maximum number of features to select

impl<E> SelectFromModel<E, Trained>

pub fn get_support(&self) -> SklResult<Array1<bool>>

Get the support mask

pub fn threshold(&self) -> SklResult<f64>

Get the threshold used for selection

Trait Implementations

impl<E: Clone, State: Clone> Clone for SelectFromModel<E, State>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<E: Debug, State: Debug> Debug for SelectFromModel<E, State>

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

Formats the value using the given formatter.

impl<E> Estimator for SelectFromModel<E, Untrained>

type Config = ()

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) -> &Self::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<E> Fit<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>, f64>> for SelectFromModel<E, Untrained>

where E: Clone + Fit<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>, f64>> + Send + Sync, E::Fitted: FeatureImportance + Send + Sync,

type Fitted = SelectFromModel<E, Trained>

The fitted model type

fn fit( self, x: &ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, y: &ArrayBase<OwnedRepr<f64>, Dim<[usize; 1]>, f64>, ) -> SklResult<Self::Fitted>

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<E> Fit<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, ArrayBase<OwnedRepr<i32>, Dim<[usize; 1]>, i32>> for SelectFromModel<E, Untrained>

where E: Clone + Fit<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, ArrayBase<OwnedRepr<i32>, Dim<[usize; 1]>, i32>> + Send + Sync, E::Fitted: FeatureImportance + Send + Sync,

type Fitted = SelectFromModel<E, Trained>

The fitted model type

fn fit( self, x: &ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, y: &ArrayBase<OwnedRepr<i32>, Dim<[usize; 1]>, i32>, ) -> SklResult<Self::Fitted>

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<E> Fit<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, ArrayBase<OwnedRepr<usize>, Dim<[usize; 1]>, usize>> for SelectFromModel<E, Untrained>

where E: Clone + Fit<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, ArrayBase<OwnedRepr<usize>, Dim<[usize; 1]>, usize>> + Send + Sync, E::Fitted: FeatureImportance + Send + Sync,

type Fitted = SelectFromModel<E, Trained>

The fitted model type

fn fit( self, x: &ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>, f64>, y: &ArrayBase<OwnedRepr<usize>, Dim<[usize; 1]>, usize>, ) -> SklResult<Self::Fitted>

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<E> Transform<ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>> for SelectFromModel<E, Trained>

fn transform(&self, x: &Array2<Float>) -> SklResult<Array2<Float>>

Transform the input data