Struct TargetEncoder 

pub struct TargetEncoder {
    pub smoothing: f32,
    pub min_samples_leaf: usize,
    /* private fields */
}

Target Encoder

Encodes categorical features using target statistics. Useful for high-cardinality categorical features.

Fields

smoothing: f32

min_samples_leaf: usize

Implementations

impl TargetEncoder

pub fn new() -> Self

pub fn smoothing(self, value: f32) -> Self

pub fn min_samples_leaf(self, value: usize) -> Self

pub fn fit(&mut self, categories: &[String], target: &[f32])

Fit encoder on categorical features and target

pub fn transform(&self, categories: &[String]) -> Vec<f32>

Transform categories to encoded values

pub fn fit_transform( &mut self, categories: &[String], target: &[f32], ) -> Vec<f32>