Struct LabelEncoder 

pub struct LabelEncoder { /* private fields */ }

Encode target labels with values between 0 and n_classes - 1, mirroring sklearn.preprocessing.LabelEncoder. Classes are sorted (sklearn default).

Operates on Vec<String> (1-D) input.

Implementations

impl LabelEncoder

pub fn new() -> Self

Creates a new label encoder.

pub fn handle_unknown(self, val: LabelHandleUnknown) -> Self

Set how unknown labels are handled during transform.

pub fn classes(&self) -> &[String]

Returns the sorted class labels.

pub fn fit<I, S>(&mut self, labels: I) -> Result<()>

where I: IntoIterator<Item = S>, S: Into<String>,

Learns the sorted class labels from the input.

pub fn transform<I>(&self, labels: I) -> Result<Vec<usize>>

where I: IntoIterator, I::Item: AsRef<str>,

Maps each label to its class index.

pub fn fit_transform<I, S>(&mut self, labels: I) -> Result<Vec<usize>>

where I: IntoIterator<Item = S>, S: Into<String>,

Fits the encoder and transforms the labels in one step.

pub fn inverse_transform(&self, indices: &[usize]) -> Result<Vec<String>>

Maps class indices back to their labels.

Trait Implementations

impl Clone for LabelEncoder

fn clone(&self) -> LabelEncoder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LabelEncoder

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

Formats the value using the given formatter.

impl Default for LabelEncoder

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for LabelEncoder

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Estimator for LabelEncoder

impl LabelTransformer for LabelEncoder

fn name(&self) -> &'static str

Human-readable name of the transformer.

fn fit(&mut self, x: &[String]) -> Result<()>

Fit on label data, learning the unique sorted classes.

fn transform(&self, x: &[String]) -> Result<Vec<usize>>

Transform labels to integer indices.

fn inverse_transform(&self, x: &[usize]) -> Result<Vec<String>>

Reverses the transformation: indices back to strings.

fn is_fitted(&self) -> bool

Whether the transformer has been fitted.

fn fit_transform(&mut self, x: &[String]) -> Result<Vec<usize>>

Convenience: fit then transform.

impl Serialize for LabelEncoder

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.