Struct MutualInformationSelector 

pub struct MutualInformationSelector {
    pub n_features_to_select: usize,
    pub n_bins: usize,
}

Parameters for MutualInformationSelector (unfitted state).

Selects the top-k features by mutual information with the target variable. Mutual information measures the dependency between two variables: a higher score means the feature is more informative about the target.

For continuous features, values are discretized into equal-width bins before computing mutual information. The number of bins can be tuned via n_bins.

This is a supervised feature selector and requires target labels y.

Example

use anofox_ml_preprocessing::MutualInformationSelector;
use anofox_ml_core::{Fit, Transform};
use ndarray::array;

let x = array![
    [1.0, 100.0],
    [2.0, 200.0],
    [1.0, 300.0],
    [2.0, 400.0],
];
let y = array![0.0, 1.0, 0.0, 1.0]; // perfectly correlated with col 0

let selector = MutualInformationSelector::new(1);
let fitted = Fit::<f64>::fit(&selector, &x, &y).unwrap();
let x_selected = fitted.transform(&x).unwrap();

assert_eq!(x_selected.ncols(), 1);

Fields

n_features_to_select: usize

Number of top features to select.

n_bins: usize

Number of equal-width bins for discretizing continuous features.

Implementations

impl MutualInformationSelector

pub fn new(n_features_to_select: usize) -> MutualInformationSelector

Create a new selector that keeps the top n_features_to_select features.

pub fn with_n_bins(self, n_bins: usize) -> MutualInformationSelector

Set the number of bins for discretizing continuous features.

Trait Implementations

impl Clone for MutualInformationSelector

fn clone(&self) -> MutualInformationSelector

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MutualInformationSelector

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for MutualInformationSelector

fn deserialize<__D>( __deserializer: __D, ) -> Result<MutualInformationSelector, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl<F> Fit<F> for MutualInformationSelector

where F: Float,

type Fitted = FittedMutualInformationSelector<F>

fn fit( &self, x: &ArrayBase<OwnedRepr<F>, Dim<[usize; 2]>>, y: &ArrayBase<OwnedRepr<F>, Dim<[usize; 1]>>, ) -> Result<<MutualInformationSelector as Fit<F>>::Fitted, RustMlError>

impl Serialize for MutualInformationSelector

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.