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) -> Self

Create a new selector that keeps the top n_features_to_select features.

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

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

Formats the value using the given formatter.

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

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl<F: Float> Fit<F> for MutualInformationSelector

type Fitted = FittedMutualInformationSelector<F>

fn fit(&self, x: &Array2<F>, y: &Array1<F>) -> Result<Self::Fitted>

impl Serialize for MutualInformationSelector

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

where __S: Serializer,

Serialize this value into the given Serde serializer.