Struct OneHotEncoder 

#[non_exhaustive]
pub struct OneHotEncoder { /* private fields */ }

One-hot encoder for integer-encoded categorical features.

Replaces each selected column with n_categories binary columns (minus any dropped by the DropStrategy). Non-selected columns pass through untouched.

Implementations

impl OneHotEncoder

pub fn new(feature_indices: Vec<usize>) -> Self

Create a new encoder for the given feature column indices.

pub fn drop(self, strategy: DropStrategy) -> Self

Set the drop strategy.

pub fn handle_unknown(self, strategy: UnknownStrategy) -> Self

Set the unknown-category strategy.

pub fn categories(&self) -> &[Vec<f64>]

Learned categories for each encoded feature.

pub fn get_feature_names(&self) -> Vec<String>

Compute the output feature names that transform would produce.

Trait Implementations

impl Clone for OneHotEncoder

fn clone(&self) -> OneHotEncoder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OneHotEncoder

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

Formats the value using the given formatter.

impl Transformer for OneHotEncoder

fn fit(&mut self, data: &Dataset) -> Result<()>

Learn parameters from the training data.

fn transform(&self, data: &mut Dataset) -> Result<()>

Apply the learned transformation to a dataset (in-place).

fn inverse_transform(&self, data: &mut Dataset) -> Result<()>

Reverse the transformation (if invertible).

fn fit_transform(&mut self, data: &mut Dataset) -> Result<()>

Convenience: fit + transform in one call.