Struct MutualInfoSelector 

pub struct MutualInfoSelector { /* private fields */ }

Mutual information based feature selection

Selects features based on mutual information between features and target.

Implementations

impl MutualInfoSelector

pub fn new(k: usize) -> Self

Create a new mutual information selector

Arguments

• k - Number of top features to select

pub fn with_discrete_target(self) -> Self

Use discrete mutual information (for classification)

pub fn with_n_neighbors(self, nneighbors: usize) -> Self

Set number of neighbors for KNN estimation

pub fn fit(&mut self, x: &Array2<f64>, y: &Array1<f64>) -> Result<()>

Fit the selector

pub fn transform(&self, x: &Array2<f64>) -> Result<Array2<f64>>

Transform data by selecting features

pub fn fit_transform( &mut self, x: &Array2<f64>, y: &Array1<f64>, ) -> Result<Array2<f64>>

Fit and transform in one step

pub fn get_support(&self) -> Option<&Vec<usize>>

Get selected feature indices

pub fn scores(&self) -> Option<&Array1<f64>>

Get mutual information scores

Trait Implementations

impl Clone for MutualInfoSelector

fn clone(&self) -> MutualInfoSelector

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MutualInfoSelector

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

Formats the value using the given formatter.