Struct OrdinalEncoder

pub struct OrdinalEncoder { /* private fields */ }

OrdinalEncoder for converting categorical features to ordinal integers

This transformer converts categorical features into ordinal integers, where each category is assigned a unique integer.

Implementations

impl OrdinalEncoder

pub fn new(handle_unknown: &str, unknown_value: Option<f64>) -> Result<Self>

Creates a new OrdinalEncoder

Arguments

• handle_unknown - How to handle unknown categories (‘error’ or ‘use_encoded_value’)
• unknown_value - Value to use for unknown categories (when handle_unknown=‘use_encoded_value’)

Returns

• A new OrdinalEncoder instance

pub fn with_defaults() -> Self

Creates an OrdinalEncoder with default settings

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

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

Fits the OrdinalEncoder to the input data

Arguments

• x - The input categorical 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 OrdinalEncoder

Arguments

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

Returns

• Result<Array2<f64>> - The ordinally encoded data

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

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

Fits the OrdinalEncoder to the input data and transforms it

Arguments

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

Returns

• Result<Array2<f64>> - The ordinally encoded data

pub fn categories(&self) -> Option<&Vec<Vec<u64>>>

Returns the categories for each feature

Returns

• Option<&Vec<Vec<u64>>> - The categories for each feature