Struct SimpleImputer

pub struct SimpleImputer { /* private fields */ }

SimpleImputer for filling missing values

This transformer fills missing values using simple strategies like mean, median, most frequent value, or a constant value.

Implementations

impl SimpleImputer

pub fn new(strategy: ImputeStrategy, missing_values: f64) -> Self

Creates a new SimpleImputer

Arguments

• strategy - The imputation strategy to use
• missing_values - The value that represents missing data (default: NaN)

Returns

• A new SimpleImputer instance

pub fn with_strategy(strategy: ImputeStrategy) -> Self

Creates a SimpleImputer with NaN as missing value indicator

Arguments

• strategy - The imputation strategy to use

Returns

• A new SimpleImputer instance

pub fn fit<S>(&mut self, x: &ArrayBase<S, Ix2>) -> Result<()>

where S: Data, S::Elem: Float + NumCast,

Fits the SimpleImputer to the input data

Arguments

• x - The input data, shape (n_samples, n_features)

Returns

• Result<()> - Ok if successful, Err otherwise

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

where S: Data, S::Elem: Float + NumCast,

Transforms the input data using the fitted SimpleImputer

Arguments

• x - The input data, shape (n_samples, n_features)

Returns

• Result<Array2<f64>> - The transformed data with imputed values

pub fn fit_transform<S>(&mut self, x: &ArrayBase<S, Ix2>) -> Result<Array2<f64>>

where S: Data, S::Elem: Float + NumCast,

Fits the SimpleImputer to the input data and transforms it

Arguments

• x - The input data, shape (n_samples, n_features)

Returns

• Result<Array2<f64>> - The transformed data with imputed values

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

Returns the statistics computed during fitting

Returns

• Option<&Array1<f64>> - The statistics for each feature