Struct MeanImputer 

pub struct MeanImputer { /* private fields */ }

Replaces NaN values with the per-feature running mean.

When a feature has seen zero non-NaN values, NaN is replaced with 0.0. This transformer accepts NaN in its input.

Implementations

impl MeanImputer

pub fn new(feature_count: usize) -> Result<Self, RillError>

Create a new imputer for feature_count features.

Errors

Returns RillError::EmptyFeatures if feature_count is 0.

pub fn means(&self) -> &[f64]

The per-feature running means of observed non-NaN values.

pub fn counts(&self) -> &[u64]

The per-feature counts of observed non-NaN values.

Trait Implementations

impl Clone for MeanImputer

fn clone(&self) -> MeanImputer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MeanImputer

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

Formats the value using the given formatter.

impl Transformer for MeanImputer

fn input_dim(&self) -> usize

Expected number of input features.

fn output_dim(&self) -> usize

Number of features produced by transform.

fn transform(&self, features: &[f64]) -> Result<Vec<f64>, RillError>

Transform features using the current internal state.

fn update(&mut self, features: &[f64]) -> Result<(), RillError>

Update internal statistics using raw features.

fn samples_seen(&self) -> u64

How many samples the transformer has seen.

fn reset(&mut self)

Reset the transformer to its initial state.